使用Python和Keras库实现基于双向门控循环单元(BiGRU)模型进行深度学习序列预测的示例

下面是一个使用Python和Keras库实现基于双向门控循环单元(BiGRU)模型进行深度学习序列预测的示例。我们将使用一个简单的正弦波时间序列数据来演示该过程。

步骤:

  1. 数据生成:生成正弦波时间序列数据。
  2. 数据预处理:将数据划分为训练集和测试集,并将其转换为适合模型输入的格式。
  3. 模型构建:构建BiGRU模型。
  4. 模型训练:使用训练数据对模型进行训练。
  5. 模型评估:使用测试数据评估模型的性能。
  6. 预测可视化:可视化模型的预测结果。

代码实现:

python 复制代码
import numpy as np
import matplotlib.pyplot as plt
from tensorflow.keras.models import Sequential
from tensorflow.keras.layers import Bidirectional, GRU, Dense
from sklearn.preprocessing import MinMaxScaler

# 1. 数据生成
def generate_sine_wave(seq_length):
    time = np.linspace(0, 2 * np.pi, seq_length)
    sine_wave = np.sin(time)
    return sine_wave

# 2. 数据预处理
def prepare_data(data, look_back):
    X, y = [], []
    for i in range(len(data) - look_back):
        X.append(data[i:(i + look_back)])
        y.append(data[i + look_back])
    X = np.array(X)
    y = np.array(y)
    # 调整输入数据的形状以适应BiGRU模型 [样本数, 时间步长, 特征数]
    X = np.reshape(X, (X.shape[0], X.shape[1], 1))
    return X, y

# 生成正弦波数据
seq_length = 1000
sine_wave = generate_sine_wave(seq_length)

# 数据归一化
scaler = MinMaxScaler(feature_range=(0, 1))
sine_wave = scaler.fit_transform(sine_wave.reshape(-1, 1)).flatten()

# 划分训练集和测试集
train_size = int(len(sine_wave) * 0.8)
train_data = sine_wave[:train_size]
test_data = sine_wave[train_size:]

# 准备训练数据和测试数据
look_back = 10
X_train, y_train = prepare_data(train_data, look_back)
X_test, y_test = prepare_data(test_data, look_back)

# 3. 模型构建
model = Sequential()
model.add(Bidirectional(GRU(50, activation='relu', input_shape=(look_back, 1))))
model.add(Dense(1))
model.compile(optimizer='adam', loss='mse')

# 4. 模型训练
model.fit(X_train, y_train, epochs=50, batch_size=32, verbose=1)

# 5. 模型评估
test_loss = model.evaluate(X_test, y_test, verbose=0)
print(f'Test Loss: {test_loss}')

# 6. 预测可视化
train_predict = model.predict(X_train)
test_predict = model.predict(X_test)

# 反归一化
train_predict = scaler.inverse_transform(train_predict)
y_train = scaler.inverse_transform(y_train.reshape(-1, 1))
test_predict = scaler.inverse_transform(test_predict)
y_test = scaler.inverse_transform(y_test.reshape(-1, 1))

# 绘制训练集预测结果
plt.figure(figsize=(12, 6))
plt.plot(y_train, label='True Train Values')
plt.plot(train_predict, label='Predicted Train Values')
plt.title('Train Data Prediction')
plt.xlabel('Time Step')
plt.ylabel('Value')
plt.legend()
plt.show()

# 绘制测试集预测结果
plt.figure(figsize=(12, 6))
plt.plot(y_test, label='True Test Values')
plt.plot(test_predict, label='Predicted Test Values')
plt.title('Test Data Prediction')
plt.xlabel('Time Step')
plt.ylabel('Value')
plt.legend()
plt.show()

代码解释:

  1. 数据生成generate_sine_wave 函数生成一个正弦波时间序列数据。
  2. 数据预处理prepare_data 函数将时间序列数据转换为适合BiGRU模型输入的格式。同时,使用 MinMaxScaler 对数据进行归一化处理。
  3. 模型构建 :使用Keras的 Sequential 模型构建一个简单的BiGRU模型,包含一个双向GRU层和一个全连接层。
  4. 模型训练 :使用 fit 方法对模型进行训练,设置训练轮数为50,批次大小为32。
  5. 模型评估 :使用 evaluate 方法评估模型在测试集上的性能。
  6. 预测可视化 :使用 matplotlib 库绘制训练集和测试集的预测结果。

通过以上步骤,你可以使用BiGRU模型进行序列预测。

相关推荐
AndrewHZ5 分钟前
【图像处理】ISP(Image Signal Processor) 图像处理器的用途和工作原理?
图像处理·人工智能·深度学习·算法·智能手机·影像系统·isp
电脑玩家粉色男孩26 分钟前
爬虫(requsets)笔记
笔记·爬虫·python
Sendingab31 分钟前
3.5 Spring Boot邮件服务:从基础发送到模板邮件进阶
spring boot·后端·python
杜子腾dd40 分钟前
17.使用读写包操作Excel文件:pyxlsb 包
python·数据挖掘·excel·numpy·pandas
夏末之花1 小时前
免费看付费电影网站制作,高清电影集合搜索引擎网站
python
JeJe同学1 小时前
将 YOLO 格式的标注文件(.txt)转换为 VOC 格式的 XML 标注文件
xml·深度学习·yolo
Python破壁人手记2 小时前
《我的Python觉醒之路》之转型Python(十三)——控制流
开发语言·python·神经网络·学习·机器学习
dundunmm2 小时前
联邦学习(Federated Learning)
人工智能·深度学习·安全·数据挖掘·联邦学习
eqwaak02 小时前
实时数仓中的Pandas:基于Flink+Arrow的流式处理方案——毫秒级延迟下的混合计算新范式
大数据·分布式·python·学习·flink·pandas
带娃的IT创业者3 小时前
《Python实战进阶》No23: 使用 Selenium 自动化浏览器操作
python·selenium·自动化