From 1be83471bc77540823a44dac7fb72165bfff61c6 Mon Sep 17 00:00:00 2001 From: danielstankw Date: Thu, 28 Sep 2023 13:03:20 +0000 Subject: [PATCH] removed - sign from f1 --- pipeline/training_pipeline.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pipeline/training_pipeline.py b/pipeline/training_pipeline.py index a8bc537..d755c84 100644 --- a/pipeline/training_pipeline.py +++ b/pipeline/training_pipeline.py @@ -283,7 +283,7 @@ def objective(trial): mlflow.log_artifact("model/preprocessor.b", artifact_path="artifact") mlflow.xgboost.log_model(model, artifact_path="model", signature=signature) - return -f1 + return f1 # Create an Optuna study and optimize the objective function study = optuna.create_study(direction="maximize")