Skip to content
Snippets Groups Projects
Commit d89d9d52 authored by Charly LAMOTHE's avatar Charly LAMOTHE
Browse files

Remove two useless concatenate in trainer.py to save space and time

parent d67e49c8
No related branches found
No related tags found
1 merge request!3clean scripts
Hide whitespace changes
Inline Side-by-side
code/bolsonaro/trainer.py
+ 2
− 2
View file @ d89d9d52
...@@ -25,9 +25,9 @@ class Trainer(object): ...@@ -25,9 +25,9 @@ class Trainer(object):
self._logger.debug('Fitting the forest on train subset and OMP on dev subset.') self._logger.debug('Fitting the forest on train subset and OMP on dev subset.')
else: else:
X_forest = np.concatenate([self._dataset.X_train, self._dataset.X_dev]) 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_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.') self._logger.debug('Fitting both the forest and OMP on train+dev subsets.')
model.fit( model.fit(
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment