TensorBoard(3)模型图

【摘要】 TensorBoard的图仪表盘(graph)是检测模型的强大工具,可以快速查看模型结构的预览图,并确保其符合预期想法。还可以查看操作级图,以了解Tensorflow程序。

前言

TensorBoard的图仪表盘(graph)是检测模型的强大工具,可以快速查看模型结构的预览图,并确保其符合预期想法。还可以查看操作级图,以了解Tensorflow程序。

一、训练模型并记录数据

首先定义一个简单的顺序模型

# Define the model.
model = keras.models.Sequential([
    keras.layers.Flatten(input_shape=(28, 28)),
    keras.layers.Dense(32, activation='relu'),
    keras.layers.Dropout(0.2),
    keras.layers.Dense(10, activation='softmax')
])

model.compile(
    optimizer='adam',
    loss='sparse_categorical_crossentropy',
    metrics=['accuracy'])

© 版权声明
THE END
喜欢就支持一下吧
点赞0 分享