为什么 Keras 中的 plot_model 不能正确绘制模型？

Question

我想可视化我的神经网络。因此，我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)

但是，当我打开图形时，它看起来像这样：

我的代码有什么问题？我没有看到任何错误。

Answer 1

原因是模型还没有建立，因为它不知道它的输入形状。使用input_shape(or input_dim) 参数在第一层指定模型的输入形状，或者通过调用fit方法开始在某些数据上拟合模型（因此可以自动推断输入形状）。此外，正如@xdurch0 在评论部分所提到的，另一种选择是调用build模型的方法并将输入形状作为参数传递给它。