import numpy as np
import matplotlib.pyplot as plt
training_epochs=20
#定义loss/accuracy一维数组
listloss = np.zeros([training_epoch])
listacc = np.zeros([training_epoch])
#填充训练得到的数据
listloss[i] = sess.run(cross_entropy, feed_dict={x_data: batch_xs, y_data: batch_ys})
listccuracy[i] = sess.run(accuracy, feed_dict={x_data: batch_xs, y_data: batch_ys})
# 绘制曲线在同一个画布上
epochs = np.arange(1, training_epochs + 1)
fig, ax1 = plt.subplots()
ax2 = ax1.twinx()
lns1 = ax1.plot(epochs, listacc, label="acc")
lns2 = ax2.plot(epochs, listloss, 'r', label="cost")
ax1.set_xlabel('iteration')
ax1.set_ylabel('training accuracy')
ax2.set_ylabel('training loss')
# 合并图例
lns = lns1 + lns2
labels = ["Accuracy","Loss"]
plt.title("acc and cost")
plt.legend(lns, labels, loc=7)
plt.show()
#plt.plot(epochs,test_acc,label="acc")
#plt.plot(epochs,avg_cost,label="cost")
#plt.title("acc and cost")
#plt.show()
tensorflow画神经网络或者卷积网络的loss/ acc的折线图
最新推荐文章于 2022-09-07 21:13:14 发布