Update ignite/contrib/metrics/roc_auc.py

pytorch · Jul 7, 2023 · d9f328f · d9f328f
1 parent 9519e9d
commit d9f328f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/ignite/contrib/metrics/roc_auc.py b/ignite/contrib/metrics/roc_auc.py
@@ -134,11 +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()])
-
         .. testoutput::
 
             FPR [0.0, 0.333, 0.333, 1.0]
             TPR [0.0, 0.0, 1.0, 1.0]
+            Thresholds [inf, 1.0, 0.711, 0.047]
+            TPR [0.0, 0.0, 1.0, 1.0]
             Thresholds [2.0, 1.0, 0.711, 0.047]
 
     ..  versionchanged:: 0.4.11
@@ -192,3 +193,4 @@ def compute(self) -> Tuple[torch.Tensor, torch.Tensor, torch.Tensor]:  # type: i
             thresholds = idist.broadcast(thresholds, src=0, safe_mode=True)
 
         return fpr, tpr, thresholds
+