LSTM模块的相关代码实现/缝合模块/即插即用模块

代码详解:

  1. 生成正弦波数据generate_sin_wave函数生成一个正弦波数据序列。
  2. 数据预处理 :将生成的数据转换为DataFrame,并创建输入序列和目标值。 在 X.append(df["Value"][i:i+seq_length].values) 确保提取的是序列的值。
  3. 数据形状调整:调整输入数据的形状,以符合LSTM的输入要求。
  4. 构建LSTM模型:使用Keras构建LSTM模型,并编译模型。
  5. 拆分训练和测试数据:将数据集拆分为训练集和测试集。
  6. 训练模型:使用训练数据训练模型,并在验证集上验证模型。
  7. 预测和可视化:使用模型进行预测,并绘制真实值和预测值的对比图。
复制代码
完整代码

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
from tensorflow.keras.models import Sequential
from tensorflow.keras.layers import LSTM, Dense

# 生成正弦波数据
def generate_sin_wave(seq_length, num_sequences):
    x = np.linspace(0, 100, seq_length * num_sequences)
    y = np.sin(x)
    return y

seq_length = 50  # 每个序列的长度
num_sequences = 100  # 序列数量

data = generate_sin_wave(seq_length, num_sequences)

# 转换为 DataFrame
df = pd.DataFrame(data, columns=["Value"])

# 创建输入和输出
X = []
y = []

for i in range(len(df) - seq_length):
    X.append(df["Value"][i:i+seq_length].values)  # 改动在这里,确保提取的是值
    y.append(df["Value"][i+seq_length])

X = np.array(X)
y = np.array(y)

# 数据形状调整
X = np.reshape(X, (X.shape[0], X.shape[1], 1))

# 构建LSTM模型
model = Sequential()
model.add(LSTM(50, return_sequences=True, input_shape=(seq_length, 1)))
model.add(LSTM(50))
model.add(Dense(1))

model.compile(optimizer='adam', loss='mean_squared_error')

# 打印模型结构
model.summary()

# 拆分训练和测试数据
train_size = int(len(X) * 0.8)
X_train, X_test = X[:train_size], X[train_size:]
y_train, y_test = y[:train_size], y[train_size:]

# 训练模型
model.fit(X_train, y_train, epochs=20, batch_size=32, validation_data=(X_test, y_test))

# 预测
predictions = model.predict(X_test)

# 绘制结果
plt.figure(figsize=(12, 6))
plt.plot(y_test, label='True Values')
plt.plot(predictions, label='Predictions')
plt.legend()
plt.show()
相关推荐
王上上1 天前
【论文阅读41】-LSTM-PINN预测人口
论文阅读·人工智能·lstm
19891 天前
【零基础学AI】第26讲:循环神经网络(RNN)与LSTM - 文本生成
人工智能·python·rnn·神经网络·机器学习·tensorflow·lstm
m0_678693333 天前
深度学习笔记29-RNN实现阿尔茨海默病诊断(Pytorch)
笔记·rnn·深度学习
24毕业生从零开始学ai3 天前
长短期记忆网络(LSTM):让神经网络拥有 “持久记忆力” 的神奇魔法
rnn·神经网络·lstm
.30-06Springfield4 天前
利用人名语言分类案例演示RNN、LSTM和GRU的区别(基于PyTorch)
人工智能·pytorch·python·rnn·分类·gru·lstm
suixinm14 天前
LSTM、GRU 与 Transformer网络模型参数计算
gru·lstm·transformer
是纯一呀14 天前
融合LSTM与自注意力机制的多步光伏功率预测新模型解析
人工智能·lstm·transformer·预测
AI扶我青云志15 天前
ANN、CNN、RNN 深度解析
rnn·深度学习·cnn
野生的程序媛15 天前
第三章 卷积神经网络
人工智能·rnn·深度学习·神经网络·机器学习·语言模型·cnn
weixin_5079299115 天前
R4 LSTM-火灾温度预测
人工智能·lstm