matplotlib.pyplot保存多张图片覆盖问题

1. 保存单张图片

保存单张图片代码如下:

# acc是我跑的神经网络的历代准确率
plt.plot(acc, 'r')
plt.title("Accuracy of LSTM2LSTM")
plt.xlabel("epoch")
plt.ylabel("accuracy")
plt.savefig("accCurve.png")  # 保存图片在当前目录下,名字为accCurve.png

2. 保存多张图片

跑神经网络的时候,我需要保存损失函数下降曲线和准确率上升曲线,如果按1中的代码,两者直接堆叠一起,就会导致第一张图片保存正确,但是第二张图片会和第一张重叠,或者说第二张图片绘制到第一张图片上去了,如下图所示:
在这里插入图片描述
所以需要添加一行代码,重置一下plt,防止图片覆盖,代码如下:

# loss是我的神经网络的历代损失值
plt.plot(loss, 'r')
plt.title("Loss of LSTM2LSTM")
plt.xlabel("epoch")
plt.ylabel("Loss")
plt.savefig("lossCurve.png")  # 保存图片在当前目录下,名字为accCurve.png

plt.clf()  # 添加上这一行,画完第一个图后,将plt重置

# acc是神经网络的历代准确率
plt.plot(acc, 'r')
plt.title("Accuracy of LSTM2LSTM")
plt.xlabel("epoch")
plt.ylabel("accuracy")
plt.savefig("accCurve.png")  # 保存图片在当前目录下,名字为accCurve.png

这样两张图就都没有问题了,准确率曲线如下所示:
在这里插入图片描述

评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值