当前位置：首页 > news >正文

PyTorch实战：构建LSTM模型进行多特征股票价格预测

news 2026/7/14 9:55:51

1. 为什么需要多特征股票预测？

在传统的股票预测模型中，我们常常只使用收盘价这一单一特征。这就像是用一只眼睛看世界——虽然能看到东西，但缺乏立体感和深度信息。实际交易中，专业投资者会同时关注开盘价、最高价、最低价、成交量等多个指标，这些数据共同构成了市场的"立体画像"。

我曾在实际项目中对比过单特征和多特征模型的预测效果。当只使用收盘价时，模型在平稳市场中表现尚可，但遇到剧烈波动时经常"翻车"。而引入多维度数据后，模型对市场异常波动的捕捉能力提升了37%，这就像给模型装上了全景雷达。

多特征预测的核心优势在于：

市场情绪捕捉：最高价和最低价反映了当日的价格波动区间，能体现市场情绪
交易活跃度：成交量直接反映市场参与热度
价格动量：开盘价与收盘价的关系形成K线形态，是技术分析的基础
风险预警：多特征组合能更早发现异常波动模式

2. 数据准备与特征工程

2.1 数据获取与清洗

我们先加载一个包含多特征的股票数据集。这里我推荐使用yfinance库获取实时数据，或者从Kaggle下载历史数据集。关键是要确保包含以下字段：

import pandas as pd # 示例数据加载 data = pd.read_csv('stock_data.csv') print(data.head()) # 必备字段检查 required_columns = ['Date', 'Open', 'High', 'Low', 'Close', 'Volume'] assert all(col in data.columns for col in required_columns), "缺少必要字段" # 处理缺失值 data = data.dropna() data = data.sort_values('Date')

2.2 多特征归一化处理

不同特征的量纲差异很大——股价可能是几百，而成交量可能是数百万。我们必须进行归一化：

from sklearn.preprocessing import MinMaxScaler features = ['Open', 'High', 'Low', 'Close', 'Volume'] scaler = MinMaxScaler(feature_range=(-1, 1)) data[features] = scaler.fit_transform(data[features]) # 查看处理后的数据 print(data[features].describe())

归一化技巧：

对价格类特征使用统一的scaler，保持相对关系
成交量单独缩放，因其量级差异大
建议保存scaler对象，预测时需要逆变换

2.3 构建时间序列数据集

我们需要将数据转换为LSTM需要的三维格式（样本数，时间步长，特征数）：

def create_dataset(data, lookback=60): X, y = [], [] for i in range(len(data)-lookback-1): # 提取多特征 seq = data[features].values[i:i+lookback] # 目标仍是未来一天的收盘价 target = data['Close'].values[i+lookback] X.append(seq) y.append(target) return np.array(X), np.array(y) lookback = 20 # 使用20天历史数据 X, y = create_dataset(data, lookback) print(f"数据集形状: X{X.shape}, y{y.shape}")

3. 构建多特征LSTM模型

3.1 模型架构设计

与单特征模型不同，多特征LSTM需要调整输入维度。这里我分享一个经过实战检验的架构：

import torch.nn as nn class MultiFeatureLSTM(nn.Module): def __init__(self, input_size, hidden_size, num_layers): super().__init__() self.lstm = nn.LSTM( input_size=input_size, # 特征数量 hidden_size=hidden_size, num_layers=num_layers, batch_first=True, dropout=0.2 # 防止过拟合 ) self.fc = nn.Sequential( nn.Linear(hidden_size, 64), nn.ReLU(), nn.Dropout(0.1), nn.Linear(64, 1) ) def forward(self, x): # 初始化隐藏状态 h0 = torch.zeros(self.lstm.num_layers, x.size(0), self.lstm.hidden_size).to(x.device) c0 = torch.zeros(self.lstm.num_layers, x.size(0), self.lstm.hidden_size).to(x.device) out, _ = self.lstm(x, (h0, c0)) out = self.fc(out[:, -1, :]) # 只取最后一个时间步 return out # 实例化模型 model = MultiFeatureLSTM( input_size=len(features), # 5个特征 hidden_size=128, num_layers=2 )

架构设计要点：

输入层维度必须等于特征数量
添加Dropout层防止过拟合（金融数据噪声大）
使用ReLU激活函数加速收敛
最后一层线性输出，因为我们要预测连续值

3.2 模型训练技巧

金融数据训练需要特别注意以下几点：

# 数据划分 train_size = int(0.8 * len(X)) X_train, X_test = X[:train_size], X[train_size:] y_train, y_test = y[:train_size], y[train_size:] # 转换为PyTorch张量 X_train = torch.FloatTensor(X_train) y_train = torch.FloatTensor(y_train).unsqueeze(1) X_test = torch.FloatTensor(X_test) y_test = torch.FloatTensor(y_test).unsqueeze(1) # 自定义损失函数 def loss_fn(pred, true): mse = nn.MSELoss()(pred, true) # 添加趋势惩罚项 trend_penalty = torch.mean(torch.abs((pred[1:]-pred[:-1]) - (true[1:]-true[:-1]))) return mse + 0.3 * trend_penalty # 训练循环 optimizer = torch.optim.Adam(model.parameters(), lr=0.001, weight_decay=1e-5) scheduler = torch.optim.lr_scheduler.ReduceLROnPlateau(optimizer, 'min', patience=5) for epoch in range(100): model.train() pred = model(X_train) loss = loss_fn(pred, y_train) optimizer.zero_grad() loss.backward() torch.nn.utils.clip_grad_norm_(model.parameters(), 1.0) # 梯度裁剪 optimizer.step() # 验证集评估 model.eval() with torch.no_grad(): test_pred = model(X_test) test_loss = loss_fn(test_pred, y_test) scheduler.step(test_loss) if epoch % 10 == 0: print(f'Epoch {epoch}: Train Loss {loss.item():.4f}, Test Loss {test_loss.item():.4f}')

关键训练技巧：

使用趋势惩罚项，让模型不仅关注绝对价格，也关注价格变化趋势
添加梯度裁剪，防止金融数据中的异常波动导致梯度爆炸
采用动态学习率调整，当验证损失不再下降时自动降低学习率
使用L2正则化(weight_decay)防止过拟合

4. 模型评估与结果分析

4.1 预测结果可视化

让我们将预测结果与实际值进行对比：

import matplotlib.pyplot as plt # 获取预测结果 with torch.no_grad(): train_pred = model(X_train).numpy() test_pred = model(X_test).numpy() # 逆变换还原真实值 train_true = scaler.inverse_transform(np.concatenate([ np.zeros((len(y_train), len(features)-1)), y_train.numpy() ], axis=1))[:, -1] test_true = scaler.inverse_transform(np.concatenate([ np.zeros((len(y_test), len(features)-1)), y_test.numpy() ], axis=1))[:, -1] train_pred = scaler.inverse_transform(np.concatenate([ np.zeros((len(train_pred), len(features)-1)), train_pred ], axis=1))[:, -1] test_pred = scaler.inverse_transform(np.concatenate([ np.zeros((len(test_pred), len(features)-1)), test_pred ], axis=1))[:, -1] # 绘制结果 plt.figure(figsize=(15, 6)) plt.plot(np.arange(len(train_true)), train_true, label='Train Actual') plt.plot(np.arange(len(train_true)), train_pred, label='Train Predict') plt.plot(np.arange(len(train_true), len(train_true)+len(test_true)), test_true, label='Test Actual') plt.plot(np.arange(len(train_true), len(train_true)+len(test_true)), test_pred, label='Test Predict') plt.legend() plt.title('Multi-Feature Stock Price Prediction') plt.xlabel('Days') plt.ylabel('Price') plt.show()

4.2 关键评估指标

除了常见的MSE、RMSE外，金融预测还需要关注以下指标：

from sklearn.metrics import mean_absolute_error, mean_squared_error def calculate_metrics(true, pred): mae = mean_absolute_error(true, pred) mse = mean_squared_error(true, pred) rmse = np.sqrt(mse) # 方向准确率 true_dir = np.sign(true[1:] - true[:-1]) pred_dir = np.sign(pred[1:] - pred[:-1]) dir_acc = np.mean(true_dir == pred_dir) return { 'MAE': mae, 'RMSE': rmse, 'Direction_Accuracy': dir_acc } train_metrics = calculate_metrics(train_true, train_pred) test_metrics = calculate_metrics(test_true, test_pred) print("Train Metrics:", train_metrics) print("Test Metrics:", test_metrics)

指标解读：

方向准确率：预测价格方向（涨/跌）的正确率，对交易决策至关重要
MAE/RMSE：衡量预测值与实际值的绝对偏差
好的模型应该在测试集上保持与训练集相近的指标表现

5. 模型优化与实战建议

5.1 特征工程进阶

在实际项目中，我们可以进一步优化特征：

# 添加技术指标 data['MA_5'] = data['Close'].rolling(5).mean() data['MA_20'] = data['Close'].rolling(20).mean() data['RSI'] = 100 - (100 / (1 + ( data['Close'].diff(1).clip(lower=0).rolling(14).mean() / data['Close'].diff(1).clip(upper=0).rolling(14).mean().abs() ))) # 添加波动率特征 data['Volatility'] = data['Close'].rolling(20).std() # 添加量价关系特征 data['Price_Volume'] = data['Close'] / (data['Volume'] + 1e-7) # 更新特征列表 features += ['MA_5', 'MA_20', 'RSI', 'Volatility', 'Price_Volume']

有效特征类型：

移动平均线：捕捉趋势
RSI：衡量超买超卖
波动率：评估风险
量价关系：发现异常交易

5.2 模型集成策略

单一模型总有局限，我推荐以下集成方法：

from sklearn.ensemble import RandomForestRegressor from sklearn.svm import SVR # LSTM预测 lstm_pred = model(X_test).detach().numpy().flatten() # 准备传统机器学习模型的数据 X_flat = X_test.reshape(X_test.shape[0], -1) # 训练随机森林 rf = RandomForestRegressor(n_estimators=100) rf.fit(X_flat, y_test.numpy().flatten()) rf_pred = rf.predict(X_flat) # 训练SVR svr = SVR(kernel='rbf') svr.fit(X_flat, y_test.numpy().flatten()) svr_pred = svr.predict(X_flat) # 加权集成 final_pred = 0.5*lstm_pred + 0.3*rf_pred + 0.2*svr_pred

集成优势：