Files
NexQuant/requirements.txt
T
TPTBusiness d12430427d feat: Add P5 ML Training Pipeline with LightGBM and 46 tests
- MLTrainer class with feature matrix builder from top factors by IC
- LightGBM training with time-series split (80/20) and early stopping
- Feature importance analysis (gain-based) with ranking
- Model persistence: model.txt + metadata.json + feature_importance.json + CSV
- Feedback generation for factor generation loop
- Model loading from disk
- Full pipeline: load factors -> train -> save -> generate feedback
- 46 unit tests covering all features
- lightgbm and scipy added to requirements.txt
2026-04-09 09:46:56 +02:00

106 lines
1.7 KiB
Plaintext

# Requirements for runtime.
pydantic-settings
python-Levenshtein
scikit-learn
filelock
loguru
fire
fuzzywuzzy
openai
litellm>=1.73 # to support `from litellm import get_valid_models`
azure.identity
pyarrow
rich
tqdm
typer
numpy # we use numpy as default data format. So we have to install numpy
pandas # we use pandas as default data format. So we have to install pandas
pandarallel # parallelize pandas
matplotlib
langchain
langchain-community
tiktoken
pymupdf # Extract shotsreens from pdf
# PDF related
pypdf
azure-ai-formrecognizer
# factor implementations
tables
# CI Fix Tool
tree-sitter-python
tree-sitter
python-dotenv
# infrastructure related.
docker
# crawler related
webdriver-manager
# demo related
streamlit>=1.47 # to support input_c.text_area(..., height="content", ...)
plotly
st-theme
randomname
flask
flask-cors
networkx
# kaggle crawler
selenium
kaggle
nbformat # also used for notebook conversion
# tool
setuptools-scm
seaborn
azure.ai.inference
# data folder desc
humanize
genson
# mlflow
mlflow
azureml-mlflow
types-pytz
# Agent
pydantic-ai-slim[mcp,openai,prefect]
nest-asyncio
# visualize SFT train
tensorboard # tensorboard --logdir git_ignore_folder/RD-Agent_workspace
prefect
# HuggingFace datasets
datasets
# DuckDuckGo search
duckduckgo-search
# Testing
pytest
pytest-cov
# Parameter Optimization
optuna>=3.5.0
# News & Data (Polymarket, ForexFactory, CryptoPanic)
beautifulsoup4>=4.12.0
# ML Training Pipeline
lightgbm>=3.3.0
scipy>=1.9.0
# RL Trading (optional - system works without these)
# Install for full RL training: pip install stable-baselines3[extra] gymnasium
# Without these, RL trading uses simple momentum fallback
# stable-baselines3[extra]>=2.0.0
# gymnasium>=0.29.0