1 import numpy as np
2 import matplotlib.pyplot as plt
3
4 traindata = np.genfromtxt("./1.txt", dtype=np.float32)[:,1] #读取.txt文件中数值
5 testdata = np.genfromtxt("./2.txt", dtype=np.float32)[:, 1]
6 plt.rcParams['font.sans-serif']=['SimHei'] #用来显示中文标签
7 plt.figure(figsize=(19.2, 10.8))
8 plt.title('损失函数衰减曲线',fontsize=25)
9 plt.xlabel("迭代次数",fontsize=20)
10 plt.ylabel("损失函数值",fontsize=20)
11 plt.plot(traindata,label="train-loss")
12 plt.plot(testdata, label="val-loss")
13 plt.tick_params(labelsize=20) #调整坐标轴刻度的字体大小
14 plt.legend(fontsize=18) #参数调整train-loss与val-loss字体的大小
15 plt.savefig("./pan1.jpg")
16 plt.show()
