Resolve "Experiment pipeline"

Closes #12 (closed)

code/bolsonaro/trainer.py

@@ -42,10 +42,14 @@ class Trainer(object):
     def base_score_metric_name(self):
         return self._base_score_metric_name
 
-    def init(self, model):
+    def init(self, model, subsets_used='train,dev'):
         if type(model) in [RandomForestRegressor, RandomForestClassifier]:
-            self._X_forest = self._dataset.X_train
-            self._y_forest = self._dataset.y_train
+            if subsets_used == 'train,dev':
+                self._X_forest = self._dataset.X_train
+                self._y_forest = self._dataset.y_train
+            else:
+                self._X_forest = np.concatenate([self._dataset.X_train, self._dataset.X_dev])
+                self._y_forest = np.concatenate([self._dataset.y_train, self._dataset.y_dev])    
             self._logger.debug('Fitting the forest on train subset')
         elif model.models_parameters.subsets_used == 'train,dev':
             self._X_forest = self._dataset.X_train