From 1bb1f5f854bad48639aff23c951d663fb68a4390 Mon Sep 17 00:00:00 2001
From: Nikolai <osterei33@gmx.de>
Date: Fri, 27 Apr 2018 10:12:14 +0200
Subject: [PATCH] Remove unnescessary call to scaler
---
toolkit.py | 8 +++-----
1 file changed, 3 insertions(+), 5 deletions(-)
diff --git a/toolkit.py b/toolkit.py
index 9955e0d..f679f69 100755
--- a/toolkit.py
+++ b/toolkit.py
@@ -263,8 +263,6 @@ class KerasROOTClassification:
if not self.data_loaded:
self._load_data()
-
- self.scaler
if not self.data_transformed:
self._transform_data()
@@ -289,14 +287,14 @@ class KerasROOTClassification:
class_weight=self.class_weight,
shuffle=True,
batch_size=self.batch_size)
-
+
logger.info("Save weights")
self.model.save_weights(os.path.join(self.project_dir, "weights.h5"))
self.total_epochs += epochs
self._write_info("epochs", self.total_epochs)
-
- logger.info("Create scores for ROC curve")
+
+ logger.info("Create scores for ROC curve")
self.scores_test = self.model.predict(self.x_test)
self.scores_train = self.model.predict(self.x_train)
--
GitLab