Update ignite/contrib/metrics/roc_auc.py

pytorch · Jul 6, 2023 · 6ac5664 · 6ac5664
1 parent 35a2da3
commit 6ac5664
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/ignite/contrib/metrics/roc_auc.py b/ignite/contrib/metrics/roc_auc.py
@@ -138,7 +138,7 @@ def sigmoid_output_transform(output):
         .. 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]
             Thresholds [2.0, 1.0, 0.711, 0.047]
 
     ..  versionchanged:: 0.4.11
@@ -192,3 +192,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
+