翻译代码 plt.figure(figsize=(15,5)) for i in range(target.size(-1)): plt.plot(target[:,:,i].numpy(), label='Target'+str(i), color='black', marker='.', linestyle='--', markersize=1, linewidth=0.5) plt.plot(range(startPoint), outSeq[:startPoint,i].numpy(), label='1-step predictions for target'+str(i), color='green', marker='.', linestyle='--', markersize=1.5, linewidth=1) # if epoch>40: # plt.plot(range(startPoint, endPoint), upperlim95[:,i].numpy(), label='upperlim'+str(i), # color='skyblue', marker='.', linestyle='--', markersize=1.5, linewidth=1) # plt.plot(range(startPoint, endPoint), lowerlim95[:,i].numpy(), label='lowerlim'+str(i), # color='skyblue', marker='.', linestyle='--', markersize=1.5, linewidth=1) plt.plot(range(startPoint, endPoint), outSeq[startPoint:,i].numpy(), label='Recursive predictions for target'+str(i), color='blue', marker='.', linestyle='--', markersize=1.5, linewidth=1) plt.xlim([startPoint-500, endPoint]) plt.xlabel('Index',fontsize=15) plt.ylabel('Value',fontsize=15) plt.title('Time-series Prediction on ' + args.data + ' Dataset', fontsize=18, fontweight='bold') plt.legend() plt.tight_layout() plt.text(startPoint-500+10, target.min(), 'Epoch: '+str(epoch),fontsize=15) save_dir = Path('result',args.data,args.filename).with_suffix('').joinpath('fig_prediction') save_dir.mkdir(parents=True,exist_ok=True) plt.savefig(save_dir.joinpath('fig_epoch'+str(epoch)).with_suffix('.png')) #plt.show() plt.close() return outSeq else: pass
时间: 2024-01-03 20:04:28 浏览: 128
“深度学习”学习日记:Tensorflow实现VGG每一个卷积层的可视化
这段代码使用了 Python 的 matplotlib 库来绘制时间序列的预测结果。其中,plt.figure() 函数设置了绘图的画布大小,for 循环用于遍历目标序列的各个维度,plt.plot() 函数用于绘制真实目标序列、一步预测结果和递归预测结果的曲线,并设置了各种绘图参数,如线条颜色、标记、线型、线宽等。plt.xlim() 函数设置了横坐标范围,plt.xlabel() 和 plt.ylabel() 函数用于设置横纵坐标的标签,plt.title() 函数设置了图像标题,plt.legend() 函数用于添加图例。plt.text() 函数在图像中添加了文本信息,指示了当前的 epoch。最后,使用 plt.savefig() 函数保存图像,并返回预测结果 outSeq。如果条件不满足,则不执行任何操作。
阅读全文