使用 keras 图启动 Tensorboard

我不明白如何使用张量板来可视化我的 keras 网络的训练步骤。


我已经用命令行启动了张量板:tensorboard --logdir=/run1 但他提出了这个错误:


当前数据集没有处于活动状态的仪表板。可能的原因:您没有将任何数据写入事件文件。TensorBoard 找不到您的事件文件。


import tensorflow as tf

from tensorflow import keras

import numpy as np


# Create the array of data


train_data = [[1.0,2.0,3.0],[4.0,5.0,6.0]]

train_data_np = np.asarray(train_data)


train_label = [[1,2,3],[4,5,6]]


train_label_np = np.asarray(train_data)


### Build the model


model = keras.Sequential([

    keras.layers.Dense(3,input_shape =(3,2)),

    keras.layers.Dense(3,activation=tf.nn.sigmoid)

])


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


#Train the model


tensorboard = TensorBoard(log_dir="run1")

model.fit(train_data_np,train_label_np,epochs=10,callbacks=tensorboard)


#test the model

restest = model.evaluate(test_data_np,test_label_np)


慕少森
浏览 225回答 1
1回答

天涯尽头无女友

在此处添加正式答案;看起来 tensorboard logdir 参数中有错字。您需要删除目录开头的斜杠tensorboard --logdir=run1
打开App,查看更多内容
随时随地看视频慕课网APP

相关分类

Python