python画图像训练结果的loss图

更新时间:2023-05-12 10:31:08 阅读: 评论:0

python画图像训练结果的loss图得到每个epoch的loss和predict精度后,就可以愉快地画图直观地看出训练结果和收敛性了。
# coding:utf-8
import matplotlib.pyplot as plt
data_dir = "D:\\"
Train_Loss_list = []
Train_Accuracy_list = []
Valid_Loss_list = []
Valid_Accuracy_list = []
f1 = open(data_dir,'r')
data = []
#把训练结果输出到⾥,⽐较笨的办法,按字节位数去取数字结果
for line in f1:
if(line.find('train')>=0):
# if (string.find(line, 'train') != -1):
Train_Loss_list.append(line[12:18])
Train_Accuracy_list.append(line[24:30])
if (line.find('valid') >= 0):
# if (string.find(line, 'valid') != -1):
Valid_Loss_list.append(line[12:18])
Valid_Accuracy_list.append(line[24:30])
f1.clo()
#迭代了30次,所以x的取值范围为(0,30),然后再将每次相对应的准确率以及损失率附在x上
x1 = range(0, 30)
x2 = range(0, 30)
y1 = Train_Accuracy_list
y2 = Train_Loss_list
y3 = Valid_Accuracy_list
y4 = Valid_Loss_list
plt.subplot(2, 1, 1)
# plt.plot(x1, y1, 'o-',color='r')
plt.plot(x1, y1, 'o-',label="Train_Accuracy")
plt.plot(x1, y3, 'o-',label="Valid_Accuracy")
plt.title('Test accuracy vs. epoches')
plt.ylabel('Test accuracy')
plt.legend(loc='best')
plt.subplot(2, 1, 2)
plt.plot(x2, y2, '.-',label="Train_Loss")
plt.plot(x2, y4, '.-',label="Valid_Loss")
plt.xlabel('Test loss vs. epoches')
plt.ylabel('Test loss')
plt.legend(loc='best')
plt.show()

本文发布于:2023-05-12 10:31:08,感谢您对本站的认可!

本文链接:https://www.wtabcd.cn/fanwen/fan/90/105589.html

版权声明:本站内容均来自互联网,仅供演示用,请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系,我们将在24小时内删除。

标签:结果   训练   字节   位数   看出
相关文章
留言与评论(共有 0 条评论)
   
验证码:
Copyright ©2019-2022 Comsenz Inc.Powered by © 专利检索| 网站地图