From 35bb6baa1a35c383d4390ea85c019ffb3ec2abbb Mon Sep 17 00:00:00 2001 From: Daniel Rosel Date: Fri, 12 Dec 2025 13:13:05 +0100 Subject: [PATCH] chore: modules and reqs --- experiments/ml/__init__.py | 11 +++++++++++ experiments/ml/requirements.txt | 6 ++++++ requirements.txt | 8 -------- 3 files changed, 17 insertions(+), 8 deletions(-) create mode 100644 experiments/ml/__init__.py create mode 100644 experiments/ml/requirements.txt 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