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()
相关推荐
木头左20 小时前
基于LSTM与3秒级Tick数据的金融时间序列预测实现
人工智能·金融·lstm
补三补四2 天前
LSTM 深度解析:从门控机制到实际应用
人工智能·rnn·lstm
孤心亦暖2 天前
RNN,GRU和LSTM的简单实现
rnn·gru·lstm
kunwen1233 天前
机器学习、深度学习
rnn·langchain·cnn·transformer·langgraph
先做个垃圾出来………5 天前
传统模型RNN与CNN介绍
人工智能·rnn·cnn
机器学习之心7 天前
MATLAB基于GM(灰色模型)与LSTM(长短期记忆网络)的组合预测方法
matlab·lstm
程序员miki7 天前
RNN循环神经网络(一):基础RNN结构、双向RNN
人工智能·pytorch·rnn·深度学习
fantasy_arch7 天前
9.3深度循环神经网络
人工智能·rnn·深度学习
机器学习之心8 天前
分解+优化+预测!CEEMDAN-Kmeans-VMD-DOA-Transformer-LSTM多元时序预测
lstm·transformer·kmeans·多元时序预测·双分解
会写代码的饭桶8 天前
通俗理解 LSTM 的三门机制:从剧情记忆到科学原理
人工智能·rnn·lstm·transformer