python 绘制损失函数曲线_python绘制loss和accuracy曲线

一、导入工具包

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&#