Update ROC_AUC Test Case in Documentation (#2989)

* Update ignite/contrib/metrics/roc_auc.py * Update roc_auc.py * Update roc_auc.py --------- Co-authored-by: sweep-ai[bot] <128439645+sweep-ai[bot]@users.noreply.github.com> Co-authored-by: vfdev <vfdev.5@gmail.com>
pytorch · Jul 10, 2023 · ac30dc3 · ac30dc3
1 parent 9519e9d
commit ac30dc3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ignite/contrib/metrics/roc_auc.py b/ignite/contrib/metrics/roc_auc.py
@@ -139,7 +139,7 @@ def sigmoid_output_transform(output):
 
             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]
+            Thresholds [inf, 1.0, 0.711, 0.047]
 
     ..  versionchanged:: 0.4.11
         added `device` argument