我做了一个非常简单的模型,所以我可以练习使用plot_model()函数,但由于某种原因我的绘图模型看起来像这样?
相比: 对方代码:
我相信您的错误来自您对“Dense”的定义:您能否展示如何导入该图层定义?
下面的代码(与您的初始代码相同,但使用 tf.keras.layers.Dense 而不是 Dense)似乎按预期工作。
将张量流导入为 tf
模型 = tf.keras.Sequential()
model.add(tf.keras.layers.Dense(10, input_shape=(1,)))
tf.keras.utils.plot_model(模型=模型,show_shapes=True)