From a2740651a3ff1630bb1979a9748050384e172d66 Mon Sep 17 00:00:00 2001 From: Roque Lopez Date: Fri, 9 Aug 2024 12:37:50 -0400 Subject: [PATCH] Add support for ROC AUC score --- alpha_automl/scorer.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/alpha_automl/scorer.py b/alpha_automl/scorer.py index 7945527a..ff33606f 100644 --- a/alpha_automl/scorer.py +++ b/alpha_automl/scorer.py @@ -7,7 +7,7 @@ from alpha_automl.utils import RANDOM_SEED, hide_logs from sklearn.metrics import accuracy_score, f1_score, jaccard_score, precision_score, recall_score,\ max_error, mean_absolute_error, mean_squared_error, mean_squared_log_error, median_absolute_error, r2_score,\ - adjusted_mutual_info_score, rand_score, mutual_info_score, normalized_mutual_info_score + adjusted_mutual_info_score, rand_score, mutual_info_score, normalized_mutual_info_score, roc_auc_score from alpha_automl.pipeline import Pipeline from alpha_automl.primitive_loader import PRIMITIVE_TYPES @@ -19,6 +19,7 @@ 'f1_score': f1_score, 'precision_score': precision_score, 'recall_score': recall_score, + 'roc_auc_score': roc_auc_score, 'jaccard_score': jaccard_score, # Regression metrics 'max_error': max_error, @@ -41,6 +42,7 @@ f1_score: 'ascending', precision_score: 'ascending', recall_score: 'ascending', + roc_auc_score: 'ascending', jaccard_score: 'ascending', # Regression metrics max_error: 'descending',