Enhance testcases for mx.metric.PCC.

apache · Jul 21, 2019 · 6a77c71 · 6a77c71
1 parent 5a67cc4
commit 6a77c71
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/tests/python/unittest/test_metric.py b/tests/python/unittest/test_metric.py
@@ -288,6 +288,13 @@ def test_pcc():
     _, pear = met_pear.get()
     np.testing.assert_almost_equal(pcc, pear)
 
+    # pcc should accept pred as scalar rather than softmax vector like
+    # Accuracy does
+    met_pcc.reset()
+    met_pcc.update(labels, [p.argmax(axis=1) for p in preds])
+    _, tst = met_pear.get()
+    np.testing.assert_almost_equal(pcc, tst)
+
     # check multiclass case against reference implementation
     CM = [
         [ 23, 13,  3 ],