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()
相关推荐
l12345sy2 天前
Day32_【 NLP _2.RNN及其变体 _(3) GRU】
rnn·自然语言处理·gru
麦麦大数据2 天前
F024 RNN+Vue+Flask电影推荐可视化系统 python flask mysql 深度学习 echarts
python·rnn·深度学习·vue·echarts·电影推荐
fsnine5 天前
从RNN到LSTM:深入理解循环神经网络与长短期记忆网络
网络·rnn·lstm
chxin140165 天前
Transformer注意力机制——动手学深度学习10
pytorch·rnn·深度学习·transformer
jie*5 天前
小杰深度学习(fourteen)——视觉-经典神经网络——ResNet
人工智能·python·深度学习·神经网络·机器学习·tensorflow·lstm
Bony-5 天前
奶茶销售数据分析
人工智能·数据挖掘·数据分析·lstm
赴3356 天前
LSTM自然语言处理情感分析项目(四)整合调用各类与方法形成主程序
人工智能·自然语言处理·lstm
赴3356 天前
LSTM自然语言处理情感分析项目(三)定义模型结构与模型训练评估测试
人工智能·自然语言处理·lstm
心动啊1217 天前
Tensorflow循环神经网络RNN
人工智能·rnn·tensorflow
清风吹过7 天前
LSTM新架构论文分享6:LSTM+Transformer融合
论文阅读·人工智能·深度学习·神经网络·lstm·transformer