Tensorboard上的损失函数

时间:2018-02-01 06:38:02

标签: python tensorflow

我是tensorflow的新手,目前我正在与Tensorboard斗争。我想在tensorboard上显示丢失功能,但我得到的只是一个空白的张量板。事件文件已创建,我也可以打开tensorboard。以下是我的代码,非常感谢任何帮助。

INFO:tensorflow:Saving dict for global step 85000: accuracy = 0.966667, global_step = 85000, loss = 0.265843

The accuracy of the model is  0.966667

The loss of the model is  0.265843

Tensorboard

代码输出:

ng2-daterangepicker

3 个答案:

答案 0 :(得分:1)

这就是:

假设损失和准确性是您想要获得的两个图表:

#Model network code

loss = z.evaluate(x=test_data.data,y=test_data.target)["loss"]
accuracy = z.evaluate(x=test_data.data,y=test_data.target)["accuracy"]

# Add summary scalar node
acc_summ = tf.summary.scalar('accuracy',accuracy)
loss_summ = tf.summary.scalar('accuracy',loss)

#Collect all summaries
merged_summary = tf.summary.merge([acc_summ, loss_summ])]

让我们看一下运行会话并将图表写入文件的代码:

summ_writer = tf.summary.FileWriter(dir_path)

# session run (To be put inside batch-loop to log loss for each batch)
summary = session.run([merged_summary], feed_dict='your input')
summ_writer.add_summary(summary, global_step)

# Finally close the writer
summ_writer.close()

答案 1 :(得分:0)

请先阅读documentation

您错过了编写图表内容的操作(如果这是您正在寻找的内容,可能会丢失)。您需要使用tf.summary.FileWriter对象来执行此操作。

答案 2 :(得分:0)

感谢大家的回复,我通过scikitlearn和tensorflow"来自" Hands on ML以及stackoverflow,最后我能够完成我的意图。如果有人感兴趣,这里是解决方案的链接: https://github.com/nish112/Using-Tendorflow-on-iris-dataset/blob/master/up_git_iris_tens.ipynb