根据txt文档绘制loss损失函数图像

txt文档中保存的数据如下:

绘制图像的代码如下:

import matplotlib.pyplot as plt

file = open('/Users/two8g/shao_work/loss.txt')  # 打开文档
data = file.readlines()  # 读取文档数据
print(data)
para_1 = []  # 新建列表,用于保存第一列数据
para_2 = []  # 新建列表,用于保存第二列数据

for num in data:
    try:
        temp = num.split(':')
        t = temp[1].split(',')
    except:
        continue
    # print(num.split(':')[1].split(' ')[0])    
    para_1.append(float(num.split(':')[1].split(' ')[0]))   
    para_2.append(float(num.split(':')[2]))
# print(para_1)

plt.figure()
plt.title('loss')
plt.plot(para_1, para_2)
plt.show()

 绘制的图像如下:

猜你喜欢

转载自blog.csdn.net/MissLemonh/article/details/120642638