def draw_acc_loss(acc_list, loss_list, epochs): """ Functions used to plot accuracy and loss values """ host = host_subplot(111) plt.subplots_adjust(right=0.8) par1 = host.twinx() host.set_xlabel("epochs") host.set_ylabel("test-loss") par1.set_ylabel("test-accuracy") p1, = host.plot(range(epochs), loss_list, label="loss") p2, = par1.plot(range(epochs), acc_list, label="accuracy") host.legend(loc=5) host.axis["left"].label.set_color(p1.get_color()) par1.axis["right"].label.set_color(p2.get_color()) host.set_xlim([0, epochs - 1]) par1.set_ylim([0, 1]) plt.draw() plt.show()
时间: 2024-04-18 07:24:30 浏览: 155
BP_Adaboost的强分类器程序设计
4星 · 用户满意度95%
这是一个用于绘制准确率和损失值曲线的函数。该函数接受三个参数:acc_list(准确率列表)、loss_list(损失值列表)和epochs(训练轮数)。
在函数内部,首先创建一个主图和一个次坐标图,用于分别显示损失值和准确率。通过调整子图的位置,使得次坐标图在主图旁边。
接下来,设置横轴标签为"epochs",纵轴标签分别为"test-loss"和"test-accuracy"。
然后,使用plot函数绘制损失值和准确率曲线。通过range(epochs)生成横轴的坐标,将loss_list和acc_list作为纵轴的数据。将损失值曲线和准确率曲线分别赋值给p1和p2。
接着,使用legend函数在图例中显示损失值曲线和准确率曲线的标签。
调整主图和次坐标图的轴标签颜色,使得与对应曲线的颜色一致。
设置横轴范围为0到epochs-1,纵轴范围为0到1。
最后,使用draw函数绘制图形,并使用show函数显示图像。
这个函数可以方便地将训练过程中的准确率和损失值可视化,帮助我们更好地理解模型的训练情况和性能表现。
阅读全文