python读取txt文件数字内容并绘svg曲线图，新罗马字体

若素不若素

已于 2022-06-22 15:33:54 修改

阅读量402

点赞数

文章标签： python 机器学习开发语言

于 2022-06-22 15:26:18 首次发布

本文链接：https://blog.csdn.net/m0_57273938/article/details/125410304

版权

这段代码读取'epoch_loss.txt'和'epoch_val_loss.txt'文件中的损失数据，将训练损失和验证损失绘制成曲线图。使用matplotlib库，设置字体为'Times New Roman'，并保存图像为'epoch_loss.png'和'epoch_loss.svg'，确保矢量图不失真。

摘要由CSDN通过智能技术生成

from matplotlib import pyplot as plt

plt.rcParams.update({
    'font.family': 'sans-serif',
    'font.sans-serif': ['Times New Roman'],
    })
losses= []
val_loss = []


with open('epoch_loss.txt', encoding='utf-8') as f:
    for line in f:
        line = line.strip('\n')
        print(line)
        losses.append(float(line))
        num=len(losses)
        iters = range(num)
with open('epoch_val_loss.txt', encoding='utf-8') as f:
    for line in f:
        line = line.strip('\n')
        val_loss.append(float(line))


plt.figure()
plt.plot(iters, losses, 'red', linewidth=2, label='train loss')
plt.plot(iters, val_loss, 'blue', linewidth=2, label='val loss')
print(iters)
print(losses)
plt.grid(True)
plt.xlabel('Epoch')
plt.ylabel('Loss')
plt.title('A Loss Curve')
plt.legend(loc="upper right")

plt.savefig("epoch_loss.png")
plt.savefig( "epoch_loss.svg")  #矢量图不失真

plt.cla()
plt.close("all")