Update classification.py

secretflow · Nov 17, 2023 · c1decdb · c1decdb
1 parent 5626595
commit c1decdb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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