Adapt to the template

prio-data · Nov 4, 2024 · 84db9b0 · 84db9b0
1 parent c6edcba
commit 84db9b0
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 8 deletions.
diff --git a/models/blank_space/src/utils/utils_run.py b/models/blank_space/src/utils/utils_run.py
@@ -96,6 +96,6 @@ def update_sweep_config(sweep_config, args, meta_config):
     sweep_config["parameters"]["name"] = {"value": meta_config["name"]}
     sweep_config["parameters"]["depvar"] = {"value": meta_config["depvar"]}
     sweep_config["parameters"]["algorithm"] = {"value": meta_config["algorithm"]}
-    if meta_config["algorithm"] == "HurdleModel":
+    if meta_config["algorithm"] == "HurdleRegression":
         sweep_config["parameters"]["model_clf"] = {"value": meta_config["model_clf"]}
         sweep_config["parameters"]["model_reg"] = {"value": meta_config["model_reg"]}
diff --git a/models/caring_fish/configs/config_hyperparameters.py b/models/caring_fish/configs/config_hyperparameters.py
@@ -10,8 +10,10 @@ def get_hp_config():
 
     hyperparameters = {
         'steps': [*range(1, 36 + 1, 1)],
-        'n_estimators': 200,
-        'tree_method': 'hist',
-        'n_jobs': 12,
+        'parameters': {
+            'n_estimators': 200,
+            'tree_method': 'hist',
+            'n_jobs': 12,
+        }
     }
     return hyperparameters
diff --git a/models/old_money/src/utils/utils_run.py b/models/old_money/src/utils/utils_run.py
@@ -64,7 +64,7 @@ def update_config(hp_config, meta_config, dp_config, args):
     config["name"] = meta_config["name"]
     config["depvar"] = meta_config["depvar"]
     config["algorithm"] = meta_config["algorithm"]
-    if meta_config["algorithm"] == "HurdleRegression":
+    if meta_config["algorithm"] == "HurdleModel":
         config["model_clf"] = meta_config["model_clf"]
         config["model_reg"] = meta_config["model_reg"]
     config["deployment_status"] = dp_config["deployment_status"]
@@ -78,6 +78,6 @@ def update_sweep_config(sweep_config, args, meta_config):
     sweep_config["parameters"]["name"] = {"value": meta_config["name"]}
     sweep_config["parameters"]["depvar"] = {"value": meta_config["depvar"]}
     sweep_config["parameters"]["algorithm"] = {"value": meta_config["algorithm"]}
-    if meta_config["algorithm"] == "HurdleRegression":
+    if meta_config["algorithm"] == "HurdleModel":
         sweep_config["parameters"]["model_clf"] = {"value": meta_config["model_clf"]}
         sweep_config["parameters"]["model_reg"] = {"value": meta_config["model_reg"]}
diff --git a/models/orange_pasta/src/utils/utils_run.py b/models/orange_pasta/src/utils/utils_run.py
@@ -64,7 +64,7 @@ def update_config(hp_config, meta_config, dp_config, args):
     config["name"] = meta_config["name"]
     config["depvar"] = meta_config["depvar"]
     config["algorithm"] = meta_config["algorithm"]
-    if meta_config["algorithm"] == "HurdleRegression":
+    if meta_config["algorithm"] == "HurdleModel":
         config["model_clf"] = meta_config["model_clf"]
         config["model_reg"] = meta_config["model_reg"]
     config["deployment_status"] = dp_config["deployment_status"]
@@ -78,6 +78,6 @@ def update_sweep_config(sweep_config, args, meta_config):
     sweep_config["parameters"]["name"] = {"value": meta_config["name"]}
     sweep_config["parameters"]["depvar"] = {"value": meta_config["depvar"]}
     sweep_config["parameters"]["algorithm"] = {"value": meta_config["algorithm"]}
-    if meta_config["algorithm"] == "HurdleRegression":
+    if meta_config["algorithm"] == "HurdleModel":
         sweep_config["parameters"]["model_clf"] = {"value": meta_config["model_clf"]}
         sweep_config["parameters"]["model_reg"] = {"value": meta_config["model_reg"]}