From 9879ac3105c5836791e6e2552404f61e06f76766 Mon Sep 17 00:00:00 2001
From: Nikolai Hartmann <Nikolai.Hartmann@physik.uni-muenchen.de>
Date: Tue, 8 May 2018 17:12:12 +0200
Subject: [PATCH] Shuffle *training* data, also for scores
---
toolkit.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/toolkit.py b/toolkit.py
index d708ee1..ee64a09 100755
--- a/toolkit.py
+++ b/toolkit.py
@@ -490,9 +490,9 @@ class KerasROOTClassification(object):
np.random.shuffle(self.y_train)
np.random.set_state(rn_state)
np.random.shuffle(self.w_train)
- if self._scores_test is not None:
+ if self._scores_train is not None:
np.random.set_state(rn_state)
- np.random.shuffle(self._scores_test)
+ np.random.shuffle(self._scores_train)
def train(self, epochs=10):
--
GitLab