diff --git a/models/blank_space/src/utils/utils_run.py b/models/blank_space/src/utils/utils_run.py index f76ffd79..9d81eff9 100644 --- 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 index 58d3fbbe..b431df6b 100644 --- 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 index f7492f52..51901a01 100644 --- 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 index f7492f52..51901a01 100644 --- 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"]}