Question

嘿，我是Tensorflow的新手。我使用DNN训练模型，我想绘制损耗曲线。但是，我不想使用Tensorboard，因为我对此并不熟悉。我想知道是否有可能在每个步骤中提取损失信息信息并使用其他绘图程序包或scikit-learn对其进行绘图？

非常感谢！

Answer 1

更改您的sess.run(training_function, feed_dict)语句，使其也包含损失函数。然后使用Matplotlib之类的东西来绘制数据。

_, loss = sess.run((training_function, loss_function), feed_dict)
loss_list.append(loss)
import matplotlib.pyplot as plt
plt.plot(loss_list)