From ae08f994d5a5bf13fba7219a41b53f9c11f52bd9 Mon Sep 17 00:00:00 2001
From: Nikolai Hartmann <Nikolai.Hartmann@physik.uni-muenchen.de>
Date: Fri, 10 Aug 2018 18:18:44 +0200
Subject: [PATCH] try to avoid copying also when fitting scaler
---
toolkit.py | 4 ++++
1 file changed, 4 insertions(+)
diff --git a/toolkit.py b/toolkit.py
index e0504e5..a3cfd20 100755
--- a/toolkit.py
+++ b/toolkit.py
@@ -19,6 +19,7 @@ import csv
import math
import glob
import shutil
+import gc
import logging
logger = logging.getLogger("KerasROOTClassification")
@@ -536,7 +537,10 @@ class ClassificationProject(object):
else:
raise ValueError("Scaler type {} unknown".format(self.scaler_type))
logger.info("Fitting {} to training data".format(self.scaler_type))
+ orig_copy_setting = self.scaler.copy
+ self.scaler.copy = False
self._scaler.fit(self.x_train, **scaler_fit_kwargs)
+ self.scaler.copy = orig_copy_setting
joblib.dump(self._scaler, filename)
return self._scaler
--
GitLab