diff --git a/src/otg/method/l2g/model.py b/src/otg/method/l2g/model.py index 23405e6b8..61deb3066 100644 --- a/src/otg/method/l2g/model.py +++ b/src/otg/method/l2g/model.py @@ -223,30 +223,7 @@ def evaluate( labelCol="label", predictionCol="prediction" ) - print("Evaluating model...") # noqa: T201 - print( # noqa: T201 - "... Area under ROC curve:", - binary_evaluator.evaluate( - results, {binary_evaluator.metricName: "areaUnderROC"} - ), - ) - print( # noqa: T201 - "... Area under Precision-Recall curve:", - binary_evaluator.evaluate( - results, {binary_evaluator.metricName: "areaUnderPR"} - ), - ) - print( # noqa: T201 - "... Accuracy:", - multi_evaluator.evaluate(results, {multi_evaluator.metricName: "accuracy"}), - ) - print( # noqa: T201 - "... F1 score:", - multi_evaluator.evaluate(results, {multi_evaluator.metricName: "f1"}), - ) - if wandb_run_name and training_data: - print("Logging to W&B...") # noqa: T201 run = wandb.init( project=self.wandb_l2g_project_name, config=hyperparameters,