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()