大家好,我是Tensorflow的新手。我使用DNN训练模型,并希望绘制损失曲线。然而,我不太熟悉Tensorboard,所以不想使用它。我想知道是否可以提取每一步的损失信息,并使用其他绘图包或scikit-learn来绘制这些数据?
非常感谢!
回答:
修改你的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)