From bd9031496df645cb4555aa975ab44bcef652657c Mon Sep 17 00:00:00 2001
From: Nikolai Hartmann <Nikolai.Hartmann@physik.uni-muenchen.de>
Date: Mon, 7 May 2018 11:23:33 +0200
Subject: [PATCH] include score name in is_train branch
---
toolkit.py | 5 ++---
1 file changed, 2 insertions(+), 3 deletions(-)
diff --git a/toolkit.py b/toolkit.py
index ec09631..70aa273 100755
--- a/toolkit.py
+++ b/toolkit.py
@@ -560,8 +560,8 @@ class KerasROOTClassification(object):
# join scores and is_train array
scores = self.evaluate(x_eval).reshape(-1)
friend_df = pd.DataFrame(np.array(scores, dtype=[(score_name, np.float64)]))
- friend_df["is_train"] = is_train
- friend_tree = friend_df.to_records()[[score_name, "is_train"]]
+ friend_df[score_name+"_is_train"] = is_train
+ friend_tree = friend_df.to_records()[[score_name, score_name+"_is_train"]]
if start == 0:
mode = "recreate"
else:
@@ -742,7 +742,6 @@ if __name__ == "__main__":
identifiers = ["DatasetNumber", "EventNumber"],
step_bkg = 100)
- c.load()
c.train(epochs=20)
c.plot_ROC()
c.plot_loss()
--
GitLab