diff --git a/experiments/ml/__init__.py b/experiments/ml/__init__.py new file mode 100644 index 0000000..11b65df --- /dev/null +++ b/experiments/ml/__init__.py @@ -0,0 +1,11 @@ +from .evals import evaluate +from .arch import ( + XGBoostAgentClassifier, + LightGBMAgentClassifier +) + +__all__ =[ + 'evaluate', + 'XGBoostAgentClassifier', + 'LightGBMAgentClassifier' +] diff --git a/experiments/ml/requirements.txt b/experiments/ml/requirements.txt new file mode 100644 index 0000000..bdbac6a --- /dev/null +++ b/experiments/ml/requirements.txt @@ -0,0 +1,6 @@ +torch +tensorboard +fastparquet +pyarrow +xgboost +lightgbm diff --git a/requirements.txt b/requirements.txt index d60d06f..e7eaf09 100644 --- a/requirements.txt +++ b/requirements.txt @@ -12,11 +12,3 @@ uv scikit-learn supabase pymc - -# ml focused -torch -tensorboard -fastparquet -pyarrow -xgboost -lightgbm