Keras plot_model未正确显示输入层

我的模型定义如下:

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))

当我使用plot_model函数绘制模型时:

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

得到的图像是 like this

其中输入层显示为一系列数字。有人知道如何让它正确显示输入层吗?


回答:

在我升级Keras后出现了这个问题

查看这个链接: https://github.com/keras-team/keras/issues/10638

在keras/engine/sequential.py中

注释掉以下代码:

@propertydef 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

Related Posts

L1-L2正则化的不同系数

我想对网络的权重同时应用L1和L2正则化。然而,我找不…

使用scikit-learn的无监督方法将列表分类成不同组别,有没有办法?

我有一系列实例,每个实例都有一份列表,代表它所遵循的不…

f1_score metric in lightgbm

我想使用自定义指标f1_score来训练一个lgb模型…

通过相关系数矩阵进行特征选择

我在测试不同的算法时,如逻辑回归、高斯朴素贝叶斯、随机…

可以将机器学习库用于流式输入和输出吗?

已关闭。此问题需要更加聚焦。目前不接受回答。 想要改进…

在TensorFlow中,queue.dequeue_up_to()方法的用途是什么?

我对这个方法感到非常困惑,特别是当我发现这个令人费解的…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注