Pytorch实现利用循环神经网络预测时间序列

最近做了利用RNN实现一个回归预测问题的小项目，在这里总结一下，方便下次使用。

代码参考莫烦python——RNN 循环神经网络 (回归)：

• https://ptorch.com/docs/4/pytorch-video-RNN-regression/
• https://morvanzhou.github.io/tutorials/machine-learning/torch/4-03-RNN-regression/

0. 构建数据集

我们的任务目的是用sin函数预测cos函数，因此首先构造数据集。这里我们用MATLAB构造数据集，并导出至csv文件。其中input为幅值1，频率5Hz的正弦信号。相应output是input求导之后的结果。

数据格式为：

t	input	output
…	…	…

1. 数据集导入

利用torch自带的Dataset和Dataloader类，导入训练数据。

导入数据集的自定义Dataset类：

import numpy