嘿,我是 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)
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)