From d6d303ec17c3b83732657e9cc421ddbb634d0216 Mon Sep 17 00:00:00 2001
From: Luc Giffon <luc.giffon@lis-lab.fr>
Date: Tue, 24 Mar 2020 09:05:07 +0100
Subject: [PATCH] delete useless main
---
code/bolsonaro/models/kmeans_forest_regressor.py | 13 +------------
1 file changed, 1 insertion(+), 12 deletions(-)
diff --git a/code/bolsonaro/models/kmeans_forest_regressor.py b/code/bolsonaro/models/kmeans_forest_regressor.py
index 591b653..551c6d8 100644
--- a/code/bolsonaro/models/kmeans_forest_regressor.py
+++ b/code/bolsonaro/models/kmeans_forest_regressor.py
@@ -121,15 +121,4 @@ class KMeansForestClassifier(KmeansForest, metaclass=ABCMeta):
def _get_best_tree_index(self, y_preds, y_true):
score = self._score_metric(y_preds, y_true)
best_tree_index = np.argmax(score) # get best scoring tree (the one with lowest mse)
- return best_tree_index
-
-if __name__ == "__main__":
- from sklearn import datasets
- from bolsonaro.models.model_parameters import ModelParameters
-
- X, y = datasets.fetch_california_housing(return_X_y=True)
- ModelParameters(extracted_forest_size=100,
- normalize_D=True,
- )
-
- k_reg = KMeansForestRegressor()
\ No newline at end of file
+ return best_tree_index
\ No newline at end of file
--
GitLab