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
这样两张图就都没有问题了,准确率曲线如下所示: