Skip to content

Commit

Permalink
Update roc_auc.py
Browse files Browse the repository at this point in the history
  • Loading branch information
vfdev-5 committed Jul 10, 2023
1 parent d9f328f commit 3ebfc9f
Showing 1 changed file with 0 additions and 3 deletions.
3 changes: 0 additions & 3 deletions ignite/contrib/metrics/roc_auc.py
Original file line number Diff line number Diff line change
Expand Up @@ -139,8 +139,6 @@ def sigmoid_output_transform(output):
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
added `device` argument
Expand Down Expand Up @@ -193,4 +191,3 @@ 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

0 comments on commit 3ebfc9f

Please sign in to comment.