From 13a859cfb8adb3a0ee658bb3184d7a3cd2800e0e Mon Sep 17 00:00:00 2001 From: vfdev Date: Mon, 10 Jul 2023 10:47:54 +0200 Subject: [PATCH] Update roc_auc.py --- ignite/contrib/metrics/roc_auc.py | 1 + 1 file changed, 1 insertion(+) diff --git a/ignite/contrib/metrics/roc_auc.py b/ignite/contrib/metrics/roc_auc.py index 4f400539c38..381e2715861 100644 --- a/ignite/contrib/metrics/roc_auc.py +++ b/ignite/contrib/metrics/roc_auc.py @@ -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]