Skip to content
Snippets Groups Projects
Commit 4c64d916 authored by Nikolai.Hartmann's avatar Nikolai.Hartmann
Browse files

fix write_friend_tree

parent aa3ffe0a
No related branches found
No related tags found
No related merge requests found
......@@ -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
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment