为什么我的plot_model()看起来像这样? (张量流、Keras)

问题描述 投票:0回答:1

我做了一个非常简单的模型,所以我可以练习使用plot_model()函数,但由于某种原因我的绘图模型看起来像这样?

相比: enter image description here 对方代码:enter image description here

python tensorflow keras deep-learning neural-network
1个回答
0
投票

我相信您的错误来自您对“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)

© www.soinside.com 2019 - 2024. All rights reserved.