From d8abe1571ccb26201cc23653344f42e00e745298 Mon Sep 17 00:00:00 2001
From: Nikolai Hartmann <Nikolai.Hartmann@physik.uni-muenchen.de>
Date: Wed, 19 Sep 2018 16:12:41 +0200
Subject: [PATCH] info message about traning stats
---
toolkit.py | 3 +++
1 file changed, 3 insertions(+)
diff --git a/toolkit.py b/toolkit.py
index be90cc6..b6ec1f9 100755
--- a/toolkit.py
+++ b/toolkit.py
@@ -1844,6 +1844,9 @@ class ClassificationProjectRNN(ClassificationProject):
x_train, y_train, w_train = self.x_train, self.y_train, self.w_train_tot
train_idx = list(self.train_val_idx[0])
np.random.seed(self.shuffle_seed+1)
+ logger.info("Generating training batches from {} signal and {} background events"
+ .format(len(np.where(self.y_train[train_idx]==1)[0]),
+ len(np.where(self.y_train[train_idx]==0)[0])))
while True:
shuffled_idx = np.random.permutation(train_idx)
for start in range(0, len(shuffled_idx), int(self.batch_size)):
--
GitLab