Skip to content

Commit 3265a9f

Browse files
committed
Add support for ROC AUC score
1 parent 68c32c5 commit 3265a9f

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

alpha_automl/scorer.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
from alpha_automl.utils import RANDOM_SEED, hide_logs
88
from sklearn.metrics import accuracy_score, f1_score, jaccard_score, precision_score, recall_score,\
99
max_error, mean_absolute_error, mean_squared_error, mean_squared_log_error, median_absolute_error, r2_score,\
10-
adjusted_mutual_info_score, rand_score, mutual_info_score, normalized_mutual_info_score
10+
adjusted_mutual_info_score, rand_score, mutual_info_score, normalized_mutual_info_score, roc_auc_score
1111
from alpha_automl.pipeline import Pipeline
1212
from alpha_automl.primitive_loader import PRIMITIVE_TYPES
1313

@@ -19,6 +19,7 @@
1919
'f1_score': f1_score,
2020
'precision_score': precision_score,
2121
'recall_score': recall_score,
22+
'roc_auc_score': roc_auc_score,
2223
'jaccard_score': jaccard_score,
2324
# Regression metrics
2425
'max_error': max_error,
@@ -41,6 +42,7 @@
4142
f1_score: 'ascending',
4243
precision_score: 'ascending',
4344
recall_score: 'ascending',
45+
roc_auc_score: 'ascending',
4446
jaccard_score: 'ascending',
4547
# Regression metrics
4648
max_error: 'descending',

0 commit comments

Comments
 (0)