From 4c64d91618ab195abeed42f197839e7b4a10f18e Mon Sep 17 00:00:00 2001
From: Nikolai Hartmann <Nikolai.Hartmann@physik.uni-muenchen.de>
Date: Tue, 14 Aug 2018 16:33:19 +0200
Subject: [PATCH] fix write_friend_tree
---
toolkit.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/toolkit.py b/toolkit.py
index 0688c12..598d60e 100755
--- a/toolkit.py
+++ b/toolkit.py
@@ -887,9 +887,9 @@ class ClassificationProject(object):
logger.info("Evaluating score for entry {}/{}".format(start, entries))
logger.debug("Loading next batch")
x_from_tree = tree2array(tree,
- branches=self.fields+self.identifiers,
+ branches=self.branches+self.identifiers,
start=start, stop=start+batch_size)
- x_eval = rec2array(x_from_tree[self.fields])
+ x_eval = rec2array(x_from_tree[self.branches])
if len(self.identifiers) > 0:
# create list of booleans that indicate which events where used for training
--
GitLab