Keras plot_model no muestra la capa de entrada apropiadamente

Mi modelo se define como tal:

model = keras.models.Sequential() model.add(layers.Embedding(max_features, 128, input_length=max_len, input_shape=(max_len,), name='embed')) model.add(layers.Conv1D(32, 7, activation='relu')) model.add(layers.MaxPooling1D(5)) model.add(layers.Conv1D(32, 7, activation='relu')) model.add(layers.GlobalMaxPooling1D()) model.add(layers.Dense(1)) 

y cuando uso la función plot_model para dibujarla:

 from keras.utils import plot_model plot_model(model, show_shapes=True, to_file='model.png') 

El dibujo que consigo es

Donde la capa de entrada es una serie de números. ¿Alguien sabe cómo dejar que muestre la entrada correctamente?

 @property def layers(self): # Historically, `sequential.layers` only returns layers that were added # via `add`, and omits the auto-generated `InputLayer` # that comes at the bottom of the stack. if self._layers and isinstance(self._layers[0], InputLayer): return self._layers[1:] return self._layers