Changing endpoint linspace

code/bolsonaro/models/model_raw_results.py

@@ -22,7 +22,7 @@ class ModelRawResults(object):
         self._test_score_base = test_score_base
         self._score_metric = score_metric
         self._base_score_metric = base_score_metric
-    
+
     @property
     def model_weights(self):
         return self.model_weights

code/train.py

@@ -208,7 +208,7 @@ if __name__ == "__main__":
     parameters['extracted_forest_size'] = np.unique(np.around(hyperparameters['n_estimators'] *
         np.linspace(0, args.extracted_forest_size_stop,
         parameters['extracted_forest_size_samples'] + 1,
-        endpoint=False)[1:]).astype(np.int)).tolist()
+        endpoint=True)[1:]).astype(np.int)).tolist()
 
     if parameters['seeds'] != None and parameters['random_seed_number'] > 1:
         logger.warning('seeds and random_seed_number parameters are both specified. Seeds will be used.')