From 5f273be5f7862321fa807f4d00093215c85a86a3 Mon Sep 17 00:00:00 2001 From: Diego Marvid Date: Tue, 5 Mar 2024 22:57:42 +0000 Subject: [PATCH] fix optuna storage default param --- pipeline_lib/implementation/tabular/xgboost/fit_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pipeline_lib/implementation/tabular/xgboost/fit_model.py b/pipeline_lib/implementation/tabular/xgboost/fit_model.py index 2a8c78b..c2c834c 100644 --- a/pipeline_lib/implementation/tabular/xgboost/fit_model.py +++ b/pipeline_lib/implementation/tabular/xgboost/fit_model.py @@ -138,7 +138,7 @@ def optuna_logging_callback(study, trial): self.logger.info(f"Optimizing XGBoost hyperparameters with {optuna_trials} trials.") study_name = optuna_params.get("study_name", "xgboost_optimization") - storage = optuna_params.get("storage", "I") + storage = optuna_params.get("storage", "sqlite:///db.sqlite3") study = optuna.create_study( direction="minimize",