python绘制loss和accuracy曲线

最新推荐文章于 2022-10-17 23:39:45 发布

Answerlzd

最新推荐文章于 2022-10-17 23:39:45 发布

阅读量9.1k

点赞数 4

CC 4.0 BY-SA版权

分类专栏： python\pytorch常用代码段文章标签： python plot 画图

本文链接：https://blog.csdn.net/Answer3664/article/details/97957038

一、导入工具包

import matplotlib.pyplot as plt
from mpl_toolkits.axes_grid1 import host_subplot

二、定义画图函数

def plot_acc_loss(loss, acc):
    host = host_subplot(111)  # row=1 col=1 first pic
    plt.subplots_adjust(right=0.8)  # ajust the right boundary of the plot window
    par1 = host.twinx()   # 共享x轴

    # set labels
    host.set_xlabel("steps")
    host.set_ylabel("test-loss")
    par1.set_ylabel("test-accuracy")

    # plot curves
    p1, = host.plot(range(len(loss)), loss, label="loss")
    p2, = par1.plot(range(len(acc)), acc, label="accuracy")

    # set location of the legend,
    # 1->rightup corner, 2->leftup corner, 3->leftdown corner
    # 4->rightdown corner, 5->rightmid ...
    host.legend(loc=5)

    # set label color
    host.axis["left"].label.set_color(p1.get_color())
    par1.axis["right"].label.set_color(p2.get_color())

    # set the range of x axis of host and y axis of par1