我正在尝试使用 TensorFlow2 构建一些模型,因此我创建了一个模型类,如下所示:
import tensorflow as tf
class Dummy(tf.keras.Model):
def __init__(self, name="dummy"):
super(Dummy, self).__init__()
self._name = name
self.dense1 = tf.keras.layers.Dense(4, activation=tf.nn.relu)
self.dense2 = tf.keras.layers.Dense(5, activation=tf.nn.softmax)
def call(self, inputs, training=False):
x = self.dense1(inputs)
return self.dense2(x)
model = Dummy()
model.build(input_shape=(None,5))
现在我想绘制模型,同时使用summary()
返回我期望的结果,plot_model(model, show_shapes=True, expand_nested=True)
只返回一个带有模型名称的块。
如何返回我的模型图?