1.下面这部分代码是绘图并保存的代码
import numpy as np
import matplotlib.pyplot as plt
x=range(10)
y=[3,1,2,9,6,2,1,8,10,8]
l=plt.plot(x,y,"b--*")
plt.xlabel("epoch")
plt.ylabel("avg acc")
plt.legend()
plt.savefig("acc.png")
plt.show()
运行结果:
2.下面是把折线图保存到指定路径下的代码:
import os
def getAccLineChart(x,y,save_dir,chart_name):
l=plt.plot(x,y,"b--*")
plt.xlabel("epoch")
plt.ylabel("avg acc")
plt.legend()
if os.path.exists(save_dir)==False:
os.makedirs(save_dir)
plt.savefig(os.path.join(save_dir,chart_name))
x=range(10)
y=[3,1,2,9,6,2,1,8,10,8]
getAccLineChart(x,y,"../graph","1.png")