From d89d9d52739cbfafcab4805d1731cffeda120329 Mon Sep 17 00:00:00 2001 From: Charly LAMOTHE <lamothe.c@intlocal.univ-amu.fr> Date: Tue, 5 Nov 2019 15:31:19 +0100 Subject: [PATCH] Remove two useless concatenate in trainer.py to save space and time --- code/bolsonaro/trainer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/code/bolsonaro/trainer.py b/code/bolsonaro/trainer.py index e5ca49b..01d0a03 100644 --- a/code/bolsonaro/trainer.py +++ b/code/bolsonaro/trainer.py @@ -25,9 +25,9 @@ class Trainer(object): self._logger.debug('Fitting the forest on train subset and OMP on dev subset.') else: X_forest = np.concatenate([self._dataset.X_train, self._dataset.X_dev]) - X_omp = np.concatenate([self._dataset.X_train, self._dataset.X_dev]) + X_omp = X_forest y_forest = np.concatenate([self._dataset.y_train, self._dataset.y_dev]) - y_omp = np.concatenate([self._dataset.y_train, self._dataset.y_dev]) + y_omp = y_forest self._logger.debug('Fitting both the forest and OMP on train+dev subsets.') model.fit( -- GitLab