From 8d911afd83a99a7869aef2eb88a3fd7cc4e6a54d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?L=C3=A9o=20Bouscarrat?= <leo.bouscarrat@euranova.eu>
Date: Fri, 6 Mar 2020 11:25:55 +0100
Subject: [PATCH] Add log to find no_weight bug
---
code/bolsonaro/models/omp_forest.py | 2 +-
code/compute_results.py | 4 +++-
2 files changed, 4 insertions(+), 2 deletions(-)
diff --git a/code/bolsonaro/models/omp_forest.py b/code/bolsonaro/models/omp_forest.py
index 35c2f8f..68b3947 100644
--- a/code/bolsonaro/models/omp_forest.py
+++ b/code/bolsonaro/models/omp_forest.py
@@ -134,7 +134,7 @@ class SingleOmpForest(OmpForest):
Make all the base tree predictions
:param X: a Forest
- :return: a np.array of the predictions of the entire forest
+ :return: a np.array of the predictions of the trees selected by OMP without applyong the weight
"""
forest_predictions = np.array([tree.predict(X) for tree in self._base_forest_estimator.estimators_])
diff --git a/code/compute_results.py b/code/compute_results.py
index 5f7fac2..4ad3686 100644
--- a/code/compute_results.py
+++ b/code/compute_results.py
@@ -365,10 +365,12 @@ if __name__ == "__main__":
omp_with_params_experiment_score_metric = extract_scores_across_seeds_and_extracted_forest_sizes(
args.models_dir, args.results_dir, args.experiment_ids[2])
#omp_with_params_without_weights
- logger.info('Loading omp_with_params experiment scores...')
+ logger.info('Loading omp_no_weights experiment scores...')
omp_with_params_without_weights_train_scores, omp_with_params_without_weights_dev_scores, omp_with_params_without_weights_test_scores, _, \
omp_with_params_experiment_score_metric = extract_scores_across_seeds_and_extracted_forest_sizes(
args.models_dir, args.results_dir, args.experiment_ids[2], weights=False)
+
+ print(omp_with_params_without_weights_test_scores)
"""# base_with_params
logger.info('Loading base_with_params experiment scores 2...')
--
GitLab