From c1decdb0e06479668b9a03804a2a8393ca71fbaf Mon Sep 17 00:00:00 2001 From: tarantula-leo <54618933+tarantula-leo@users.noreply.github.com> Date: Fri, 17 Nov 2023 17:01:36 +0800 Subject: [PATCH] Update classification.py --- sml/metrics/classification/classification.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sml/metrics/classification/classification.py b/sml/metrics/classification/classification.py index 9bbd12ca..2d3882fc 100644 --- a/sml/metrics/classification/classification.py +++ b/sml/metrics/classification/classification.py @@ -214,7 +214,7 @@ def fun_score( y_true_binary, y_pred_binary = transform_binary(y_true, y_pred, i) fun_result.append(fun(y_true_binary, y_pred_binary)) elif average == 'binary': - if transform is True: + if transform: y_true_binary, y_pred_binary = transform_binary(y_true, y_pred, pos_label) else: y_true_binary, y_pred_binary = y_true, y_pred