sweeper

mmcdermott · Aug 21, 2024 · 9c542ea · 9c542ea
1 parent 0d7ed27
commit 9c542ea
Show file tree

Hide file tree

Showing 5 changed files with 19 additions and 0 deletions.
diff --git a/src/MEDS_tabular_automl/configs/model/knn_classifier.yaml b/src/MEDS_tabular_automl/configs/model/knn_classifier.yaml
@@ -29,6 +29,10 @@ model_params:
 
 hydra:
   sweeper:
+    direction: maximize
+    n_trials: 250
+    n_jobs: 25
+
     params:
       model_params.model.n_neighbors: range(1, 20)
       model_params.model.weights: choice(['uniform', 'distance'])

diff --git a/src/MEDS_tabular_automl/configs/model/logistic_regression.yaml b/src/MEDS_tabular_automl/configs/model/logistic_regression.yaml
@@ -36,6 +36,10 @@ model_params:
 
 hydra:
   sweeper:
+    direction: maximize
+    n_trials: 250
+    n_jobs: 25
+
     params:
       model_params.model.C: tag(log, interval(1e-6, 1))
       model_params.model.penalty: choice(['l1', 'l2', 'elasticnet'])

diff --git a/src/MEDS_tabular_automl/configs/model/random_forest_classifier.yaml b/src/MEDS_tabular_automl/configs/model/random_forest_classifier.yaml
@@ -34,6 +34,10 @@ model_params:
 
 hydra:
   sweeper:
+    direction: maximize
+    n_trials: 250
+    n_jobs: 25
+
     params:
       model_params.model.n_estimators: range(50, 300, 50)
       model_params.model.max_depth: choice([null, 10, 20, 30, 40, 50])

diff --git a/src/MEDS_tabular_automl/configs/model/sgd_classifier.yaml b/src/MEDS_tabular_automl/configs/model/sgd_classifier.yaml
@@ -26,6 +26,9 @@ model_params:
 
 hydra:
   sweeper:
+    direction: maximize
+    n_trials: 250
+    n_jobs: 25
     params:
       +model_params.model.alpha: tag(log, interval(1e-6, 1))
       +model_params.model.l1_ratio: interval(0, 1)

diff --git a/src/MEDS_tabular_automl/configs/model/xgboost.yaml b/src/MEDS_tabular_automl/configs/model/xgboost.yaml
@@ -30,6 +30,10 @@ model_params:
 
 hydra:
   sweeper:
+    direction: maximize
+    n_trials: 250
+    n_jobs: 25
+
     params:
       +model_params.model.eta: tag(log, interval(0.001, 1))
       +model_params.model.lambda: tag(log, interval(0.001, 1))