想输出 training 和 validation 的 accuracy 和 loss 曲线。但是在没有图形界面的linux服务器上无法使用plot和show函数。
此时,需要更改matplotlib的后端,然后将图片保存下来观看。
在 import matplotlib.plt 之前插入:
import matplotlib as mpl
mpl.use('Agg')
#...
#...
import matplotlib.plt
在对应的需要展示图片的位置
matplotlib.plt.plot(x_axis, y_axis)
matplotlib.plt.savefig("save/image.png")
提示:如果需要绘制多个图片,记得在每一个 savefig 下面接一个 close ,不然会重叠
matplotlib.plt.savefig("save/image.png")
matplotlib.plt.close()