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

发布于:2025-03-15 ⋅ 阅读:(15) ⋅ 点赞:(0)

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

步骤:

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

代码实现:

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模型进行序列预测。