为什么Keras中的plot_model不能正确绘制模型？
为什么Keras中的plot_model不能正确绘制模型？
提问于 2020-04-15 19:05:21
我想把我的神经网络可视化。因此，我使用from tensorflow.keras.utils import plot_model并像这样使用它：
model = Sequential()
model.add(Dense(8, activation="relu"))
model.add(Dense(1))
plot_model(model, to_file="model.png", show_shapes=True)但是，当我打开图形时，它看起来是这样的：
我的代码出了什么问题？我看不到任何错误。
Stack Overflow用户
回答已采纳
发布于 2020-04-15 20:43:59
原因是模型没有被构建，因为它不知道它的输入形状。可以使用input_shape (或input_dim)参数在第一层上指定模型的输入形状，也可以通过调用fit方法在某些数据上开始拟合模型(以便可以自动推断输入形状)。此外，正如@xduch0在注释部分中提到的，另一种选择是调用模型的build方法，并将输入形状作为参数传递给它。
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/61227174
复制相关文章
相似问题
腾讯云开发者
