CNN做时间序列预测_使用PyTorch进行LSTM时间序列预测

LSTM预测航班

搬运自：LSTM，感谢原作者Usman Malik 。

（顺便种草一个软件：神经网络结构绘图软件https://github.com/alexlenail/NN-SVG）

---

本节将介绍另一种常用的门控循环神经网络：长短期记忆（long short-term memory，LSTM）。它 比门控循环单元的结构稍微复杂一点。

1.1、数据集和问题定义

import 

让我们打印Seaborn库内置的所有数据集的列表：

[

让我们将数据集加载到我们的应用程序中

flight_data 

该数据集有三列：year，month，和passengers。该passengers列包含指定月份旅行旅客的总数。让我们绘制数据集的形状：

flight_data

可以看到数据集中有144行和3列，这意味着数据集包含12年的乘客旅行记录。

任务是根据前132个月来预测最近12个月内旅行的乘客人数。请记住，我们有144个月的记录，这意味着前132个月的数据将用于训练我们的LSTM模型，而模型性