Skip to content

Commit 3865b36

Browse files
Add matthews correlation coeff as evaluation metric
1 parent f2c71ed commit 3865b36

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

cobra/evaluation/evaluator.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111
from sklearn.metrics import roc_curve
1212
from sklearn.metrics import confusion_matrix
1313
from sklearn.metrics import roc_auc_score
14+
from sklearn.metrics import matthews_corrcoef
1415
from sklearn.exceptions import NotFittedError
1516

1617

@@ -109,6 +110,7 @@ def compute_scalar_metrics(y_true: np.ndarray,
109110
"precision": precision_score(y_true, y_pred_b),
110111
"recall": recall_score(y_true, y_pred_b),
111112
"F1": f1_score(y_true, y_pred_b, average=None)[1],
113+
"matthews_corrcoef": matthews_corrcoef(y_true, y_pred_b),
112114
"lift at {}".format(lift_at): np.round(Evaluator
113115
._compute_lift(
114116
y_true=y_true,

0 commit comments

Comments
 (0)