From 006b001e4d0e987b65cee5649b3ca5b6db760155 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?L=C3=A9o=20Bouscarrat?= <leo.bouscarrat@euranova.eu>
Date: Thu, 5 Mar 2020 21:17:13 +0100
Subject: [PATCH] Changing endpoint linspace
---
code/bolsonaro/models/model_raw_results.py | 2 +-
code/train.py | 2 +-
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/code/bolsonaro/models/model_raw_results.py b/code/bolsonaro/models/model_raw_results.py
index fcb4220..fc22406 100644
--- a/code/bolsonaro/models/model_raw_results.py
+++ b/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
diff --git a/code/train.py b/code/train.py
index e51514c..8a7cceb 100644
--- a/code/train.py
+++ b/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.')
--
GitLab