Update ignite/contrib/metrics/roc_auc.py

pytorch · Jul 6, 2023 · 7747a43 · 7747a43
1 parent 9a35bfa
commit 7747a43
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/ignite/contrib/metrics/roc_auc.py b/ignite/contrib/metrics/roc_auc.py
@@ -134,7 +134,12 @@ def sigmoid_output_transform(output):
             print("FPR", [round(i, 3) for i in state.metrics['roc_auc'][0].tolist()])
             print("TPR", [round(i, 3) for i in state.metrics['roc_auc'][1].tolist()])
             print("Thresholds", [round(i, 3) for i in state.metrics['roc_auc'][2].tolist()])
-            print("TPR", [round(i, 3) for i in state.metrics['roc_auc'][1].tolist()])
+
+        .. testoutput::
+
+            FPR [0.0, 0.333, 0.333, 1.0]
+            TPR [0.0, 0.0, 1.0, 1.0]
+            Thresholds [2.0, 1.0, 0.711, 0.047]
             print("Thresholds", [round(i, 3) for i in state.metrics['roc_auc'][2].tolist()])
 
     ..  versionchanged:: 0.4.11