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")