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

include score name in is_train branch

parent f73a70ea
No related branches found
No related tags found
No related merge requests found
......@@ -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()
......
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