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 authored Jul 10, 2023
1 parent 3ebfc9f commit 13a859c
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions ignite/contrib/metrics/roc_auc.py
Original file line number Diff line number Diff line change
Expand Up @@ -134,6 +134,7 @@ 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]
Expand Down

0 comments on commit 13a859c

Please sign in to comment.