基于Python的人工智能应用案例系列（17）：LSTM正弦波预测

概述

        本案例展示了如何使用LSTM（长短期记忆网络）来预测正弦波序列的未来值。由于正弦波具有周期性，传统的神经网络难以准确预测其上升或下降趋势，而LSTM则能够通过学习值的模式来进行更精准的预测。本案例将训练LSTM模型并预测正弦波的后续值，同时展示了如何使用该模型进行未来预测。

数据生成

        我们首先生成800个正弦波数据点，并定义每40个点为一个完整周期，因此有20个完整的周期。数据集的前760个点用于训练，最后40个点作为测试集。

# 创建并绘制正弦波数据点
t = torch.linspace(0,799,steps=800)
y = torch.sin(t*2*3.1416/40)plt.figure(figsize=(12,4))
plt.xlim(-10,801)
plt.grid(True)
plt.plot(y.numpy());

构建LSTM模型

        LSTM模型由一个LSTM层和一个全连接层组成。LSTM层的输入大小为1，隐藏层大小为50，输出大小为1。每次训练后，我们会使用最后一个窗口的训练数据来预测未来值。

class LSTM(nn.Module):def __init__(self, input_size=1, hidden_size=50, out_size=1):super().__init__()self.hidden_size = hidden_sizeself.lstm = nn.LSTM(input_size, hidden_size)self.linear = nn.Linear(hidden_size, out_size)self.hidden = (torch.zeros(1, 1, hidden_size), torch.zeros(1, 1, hidden_size))def forward(self, seq):lstm_out, self.hidden = self.lstm(seq.view(len(seq), 1, -1), self.hidden)pred = self.linear(lstm_out.view(len(seq), -1))return pred[-1]

模型训练与预测

        训练LSTM模型时，首先将序列分成重叠的窗口，每个窗口包含40个点，模型根据这些窗口数据进行预测。训练过程分为10个周期，并在每个周期后使用训练数据生成的最后一个窗口预测未来40个点。

epochs = 10
future = 40for i in range(epochs):for seq, y_train in train_data:optimizer.zero_grad()model.hidden = (torch.zeros(1, 1, model.hidden_size), torch.zeros(1, 1, model.hidden_size))y_pred = model(seq)loss = criterion(y_pred, y_train)loss.backward()optimizer.step()preds = train_set[-window_size:].tolist()for f in range(future):seq = torch.FloatTensor(preds[-window_size:])with torch.no_grad():model.hidden = (torch.zeros(1, 1, model.hidden_size), torch.zeros(1, 1, model.hidden_size))preds.append(model(seq).item())loss = criterion(torch.tensor(preds[-window_size:]), y[760:])plt.figure(figsize=(12,4))plt.xlim(700,801)plt.grid(True)plt.plot(y.numpy())plt.plot(np.arange(760,800), torch.tensor(preds[window_size:]))plt.show()

预测未来值

        在对整个数据集进行训练后，我们可以预测未来40个点。训练后的LSTM模型可以通过最后一个训练窗口生成的序列，逐步预测未来的正弦波值。

preds = y[-window_size:].tolist()for i in range(future):seq = torch.FloatTensor(preds[-window_size:])with torch.no_grad():model.hidden = (torch.zeros(1, 1, model.hidden_size), torch.zeros(1, 1, model.hidden_size))preds.append(model(seq).item())plt.figure(figsize=(12,4))
plt.xlim(-10,841)
plt.grid(True)
plt.plot(y.numpy())
plt.plot(range(800,800+future), preds[window_size:])
plt.show()

结语

        在本案例中，我们利用LSTM模型成功预测了正弦波的未来值。通过训练LSTM网络识别正弦波的周期性特征，模型不仅能够精确预测下一时刻的值，还可以延展预测多个未来值，展示了LSTM在处理时间序列数据方面的强大能力。相比于传统的神经网络，LSTM的长短期记忆结构使其能够学习数据中的长期依赖关系，对于具有周期性和趋势性的数据特别有效。

        LSTM不仅适用于正弦波等简单周期信号，在更复杂的时间序列数据中，如股票市场、能源消耗等具有时序性的实际应用中，LSTM同样能够发挥重要作用。通过本案例的学习，读者可以更好地理解LSTM的应用原理，并将其拓展到更多实际场景的预测任务中。

如果你觉得这篇博文对你有帮助，请点赞、收藏、关注我，并且可以打赏支持我！

欢迎关注我的后续博文，我将分享更多关于人工智能、自然语言处理和计算机视觉的精彩内容。

谢谢大家的支持！