近期在搞深度学习,训练的模型希望能够绘制一张趋势图来观察其训练效果。所以需要用到plt来绘图
并不需要记住特别复杂精妙的方法,对我而言只需要记住几个核心的功能,达到我观察趋势的目的即可:
主要有:
plt.figure(num=3, figsize=(8, 15))#设置fig的尺寸,按需设置
plt.subplot(211)
plt.plot(plot_data["batchindex"], moving_average(plot_data["error"]), 'r--',
plot_data_validate["batchindex"], moving_average(plot_data_validate["error"]), 'b--',
plot_data_test["batchindex"], moving_average(plot_data_test["error"]), 'g--')#根据plot_data中的数据绘制
plt.grid(True) ##增加格点
plt.axis('tight') # 坐标轴适应数据量 axis 设置坐标轴
my_y_ticks = np.arange(0, 0.6, 0.01)
plt.yticks(my_y_ticks)#设置y轴显示刻度
plt.xlabel('Minibatch number')#x轴名字
plt.ylabel('Label Prediction Error')
plt.title('with BN')#整个图的名字
plt.show()
知否知否,上述只是核心代码块,完整的还需自己根据需求补全~
如果需要更多功能的绘图函数,建议参考http://www.cnblogs.com/chaoren399/p/5792168.html