Update classification_emul.py

secretflow · Nov 17, 2023 · 61f2e19 · 61f2e19
1 parent d2c2ba0
commit 61f2e19
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sml/metrics/classification/classification_emul.py b/sml/metrics/classification/classification_emul.py
@@ -95,8 +95,8 @@ def check(spu_result, sk_result):
     # Test binary
     y_true = jnp.array([0, 1, 1, 0, 1, 1])
     y_pred = jnp.array([0, 0, 1, 0, 1, 1])
-    spu_result = emulator.run(proc)(y_true, y_pred, average=None, labels=[0, 1, 2])
-    sk_result = sklearn_proc(y_true, y_pred, average=None, labels=[0, 1, 2])
+    spu_result = emulator.run(proc)(y_true, y_pred, pos_label=1, transform=0)
+    sk_result = sklearn_proc(y_true, y_pred)
     check(spu_result, sk_result)
 
     # Test multiclass