为什么KERAS中的PLOT_MODEL不能正确打印模型？ [英] Why plot_model in Keras does not plot the model correctly?

问题描述

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

但是，当我打开图形时，它看起来如下所示：

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

推荐答案

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

这篇关于为什么KERAS中的PLOT_MODEL不能正确打印模型？的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！