From 4f7d3f1f29c05bff97ec2b3db13f54da4e7f16c5 Mon Sep 17 00:00:00 2001
From: Nikolai Hartmann <Nikolai.Hartmann@physik.uni-muenchen.de>
Date: Fri, 5 Oct 2018 11:24:30 +0200
Subject: [PATCH] support for score_mode in write_friend_tree
---
toolkit.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/toolkit.py b/toolkit.py
index b6ec1f9..3e75d2c 100755
--- a/toolkit.py
+++ b/toolkit.py
@@ -1024,7 +1024,7 @@ class ClassificationProject(object):
def write_friend_tree(self, score_name,
source_filename, source_treename,
target_filename, target_treename,
- batch_size=100000):
+ batch_size=100000, score_mode=None):
f = ROOT.TFile.Open(source_filename)
tree = f.Get(source_treename)
entries = tree.GetEntries()
@@ -1050,7 +1050,7 @@ class ClassificationProject(object):
is_train = np.zeros(len(x_eval))
# join scores and is_train array
- scores = self.evaluate(x_eval).reshape(-1)
+ scores = self.evaluate(x_eval, mode=score_mode).reshape(-1)
friend_df = pd.DataFrame(np.array(scores, dtype=[(score_name, np.float64)]))
friend_df[score_name+"_is_train"] = is_train
friend_tree = friend_df.to_records()[[score_name, score_name+"_is_train"]]
--
GitLab