面向量化金融的开源机器学习模型
机器学习交易算法应用统计学习模型(梯度提升、神经网络、Transformer)于金融时间序列数据,以预测价格走势或生成交易信号。这些开源实现涵盖回归(价格目标)和分类(方向)两种表述。
ML 算法如何跨库连接
ML 算法如何在交易系统中协同工作
Price, volume, indicators
Model predictions
Signal threshold filtering
Profit taking & stop loss
Position sizing & portfolio
在关键维度上对比 ML 算法
| 指标 | LightGBMRegressorFreqtrade | LightGBMClassifierFreqtrade | XGBoostRegressorFreqtrade | XGBoostClassifierFreqtrade | CatboostRegressorFreqtrade | PyTorchMLPRegressorFreqtrade | PyTorchTransformerRegressorFreqtrade | LGBModelQlib (Microsoft) | XGBModelQlib (Microsoft) | DNNModelQlib (Microsoft) | ALSTMQlib (Microsoft) | TFTModelQlib (Microsoft) | GATsQlib (Microsoft) |
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| 复杂度 | ⭐⭐⭐intermediate | ⭐⭐⭐intermediate | ⭐⭐⭐intermediate | ⭐⭐⭐intermediate | ⭐⭐⭐intermediate | ⭐⭐⭐⭐advanced | ⭐⭐⭐⭐advanced | ⭐⭐⭐intermediate | ⭐⭐⭐intermediate | ⭐⭐⭐⭐advanced | ⭐⭐⭐⭐advanced | ⭐⭐⭐⭐advanced | ⭐⭐⭐⭐advanced |
| 预测类型 | 回归 | 分类 | 回归 | 分类 | 回归 | 回归 | 回归 | 混合 | 混合 | 混合 | 序列 | 混合 | 混合 |
| 训练速度 | ⚡⚡⚡ | ⚡⚡⚡ | ⚡⚡⚡ | ⚡⚡⚡ | ⚡⚡ | ⚡⚡ | ⚡ | ⚡⚡ | ⚡⚡ | ⚡⚡ | ⚡ | ⚡⚡ | ⚡⚡ |
| 准确度 | 📊📊📊 | 📊📊📊 | 📊📊📊 | 📊📊📊 | 📊📊📊 | 📊📊📊 | 📊📊📊📊 | 📊📊 | 📊📊 | 📊📊 | 📊📊📊 | 📊📊 | 📊📊📊📊 |
| 最适合 | 表格数据 | 表格数据 | 表格数据 | 表格数据 | 通用 | 非线性模式 | 时间序列模式 | 通用 | 通用 | 通用 | 序列数据 | 通用 | 图关系 |
Gradient boosting regression model for price movement prediction using LightGBM.
| n_estimators | 1000 | Number of boosting rounds |
| learning_rate | 0.01 | Step size shrinkage |
freqai/prediction_models/LightGBMRegressor.pyGradient boosting classification model for directional prediction (up/down/neutral).
| n_estimators | 1000 | Number of boosting rounds |
freqai/prediction_models/LightGBMClassifier.pyXGBoost-based regression model for continuous value prediction.
| n_estimators | 1000 | Number of boosting rounds |
| max_depth | 6 | Maximum tree depth |
freqai/prediction_models/XGBoostRegressor.pyXGBoost-based classification model for directional prediction.
| n_estimators | 1000 | Number of boosting rounds |
freqai/prediction_models/XGBoostClassifier.pyCatBoost gradient boosting model with native categorical feature support.
| iterations | 1000 | Number of boosting iterations |
freqai/prediction_models/CatboostRegressor.pyMulti-layer perceptron neural network for regression-based price prediction.
| hidden_dim | 128 | Hidden layer dimension |
| dropout_percent | 0.2 | Dropout rate |
freqai/prediction_models/PyTorchMLPRegressor.pyLightGBM model for stock return prediction using technical and fundamental features.
| num_leaves | 31 | Maximum number of leaves |
| learning_rate | 0.1 | Boosting learning rate |
qlib/contrib/model/gbdt.pyXGBoost model for stock return prediction.
qlib/contrib/model/xgboost.pyDeep neural network for nonlinear feature extraction and return prediction.
| hidden_size | 256 | Hidden layer size |
qlib/contrib/model/pytorch_nn.pyAttention-based LSTM for sequential stock data modeling with attention mechanism.
| hidden_size | 64 | LSTM hidden size |
| num_layers | 2 | Number of LSTM layers |
qlib/contrib/model/pytorch_alstm.pyTemporal Fusion Transformer combining static and temporal features for multi-horizon prediction.
qlib/contrib/model/pytorch_tft.py