Открытые ML-модели для количественных финансов
Торговые алгоритмы машинного обучения применяют статистические модели обучения (градиентный бустинг, нейронные сети, трансформеры) к финансовым временным рядам для прогнозирования движений цен или генерации торговых сигналов. Эти реализации с открытым исходным кодом охватывают как регрессионные (целевая цена), так и классификационные (направление) формулировки.
Как 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