Skip to content

Commit b1350ff

Browse files
Andrew LengAndrew Leng
authored andcommitted
feat: Add machine learning arbitrage prediction models
- Implement Random Forest, Gradient Boosting, XGBoost, and LightGBM models - Advanced feature engineering with market data and historical performance - Neural network predictor using PyTorch for complex patterns - Ensemble prediction combining multiple ML models - Model training, evaluation, and performance tracking - Feature importance analysis and model persistence - Real-time prediction API for arbitrage opportunities
1 parent 5f57196 commit b1350ff

File tree

2 files changed

+537
-0
lines changed

2 files changed

+537
-0
lines changed

requirements.txt

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,3 +8,9 @@ numpy>=1.21.0
88
pandas>=1.3.0
99
python-socketio>=5.0.0
1010
eventlet>=0.33.0
11+
scikit-learn>=1.0.0
12+
xgboost>=1.6.0
13+
lightgbm>=3.3.0
14+
torch>=1.12.0
15+
torchvision>=0.13.0
16+
joblib>=1.1.0

0 commit comments

Comments
 (0)