From 1c7e2228b2293f72fb91312acbb291ea6c1c8350 Mon Sep 17 00:00:00 2001
From: Thomas Weber <Thomas.Weber@physik.uni-muenchen.de>
Date: Mon, 30 Apr 2018 13:37:06 +0200
Subject: [PATCH] Write AUC into plot legends
---
compare.py | 5 +++--
toolkit.py | 3 +--
2 files changed, 4 insertions(+), 4 deletions(-)
diff --git a/compare.py b/compare.py
index 5a34371..8b6409f 100755
--- a/compare.py
+++ b/compare.py
@@ -20,9 +20,10 @@ def overlay_ROC(filename, *projects):
for p in projects:
fpr, tpr, threshold = roc_curve(p.y_test, p.scores_test, sample_weight = p.w_test)
fpr = 1.0 - fpr
-
+ roc_auc = auc(tpr, fpr)
+
plt.grid(color='gray', linestyle='--', linewidth=1)
- plt.plot(tpr, fpr, label=p.name)
+ plt.plot(tpr, fpr, label=str(p.name+" (AUC = {})".format(roc_auc)))
plt.xlim(0,1)
plt.ylim(0,1)
diff --git a/toolkit.py b/toolkit.py
index 72a4228..ccd68e7 100755
--- a/toolkit.py
+++ b/toolkit.py
@@ -499,7 +499,7 @@ class KerasROOTClassification(object):
roc_auc = auc(tpr, fpr)
plt.grid(color='gray', linestyle='--', linewidth=1)
- plt.plot(tpr, fpr, label='NN')
+ plt.plot(tpr, fpr, label=str(self.name + " (AUC = {})".format(roc_auc)))
plt.plot([0,1],[1,0], linestyle='--', color='black', label='Luck')
plt.ylabel("Background rejection")
plt.xlabel("Signal efficiency")
@@ -509,7 +509,6 @@ class KerasROOTClassification(object):
plt.xticks(np.arange(0,1,0.1))
plt.yticks(np.arange(0,1,0.1))
plt.legend(loc='lower left', framealpha=1.0)
- plt.text(0.21,0.02,"AUC: {}".format(str(roc_auc)), size=12)
plt.savefig(os.path.join(self.project_dir, "ROC.pdf"))
plt.clf()
--
GitLab