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

相关推荐
小徐敲java11 小时前
python使用s7协议与plc进行数据通讯(HslCommunication模拟)
开发语言·python
猫头虎11 小时前
如何解决 pip install 编译报错 fatal error: hdf5.h: No such file or directory(h5py)问题
人工智能·python·pycharm·开源·beautifulsoup·ai编程·pip
雪碧聊技术11 小时前
深度学习、机器学习、人工智能三者的关系
人工智能·深度学习·机器学习
p***233611 小时前
python的sql解析库-sqlparse
数据库·python·sql
陈奕昆11 小时前
n8n实战营Day1课时3:高频节点解析+Webhook表单同步Excel实操
人工智能·python·n8n
深蓝电商API11 小时前
动态 Token、加密参数逆向全流程:从原理到实战破解
爬虫·python
qq_17082750 CNC注塑机数采11 小时前
【Python TensorFlow】 TCN-GRU时间序列卷积门控循环神经网络时序预测算法(附代码)
python·rnn·神经网络·机器学习·gru·tensorflow·tcn
AI大模型学徒11 小时前
大模型应用开发(四)_调用大模型分析图片
人工智能·深度学习·ai·大模型·deepseek
java1234_小锋11 小时前
基于Python深度学习的车辆车牌识别系统(PyTorch2卷积神经网络CNN+OpenCV4实现)视频教程 - 切割车牌矩阵获取车牌字符
python·深度学习·cnn·车牌识别
陈文锦丫12 小时前
Boundary Difference Over Union Loss For Medical Image Segmentation
深度学习