diff --git a/code/bolsonaro/models/omp_forest.py b/code/bolsonaro/models/omp_forest.py index 35c2f8fa0e46e2c87a76be438a9968665e7f5b29..68b394775007946cf6259a8e259c12758012da31 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 5f7fac2c7718cf887d3d83a5b3a7eb9cdebfb9d9..4ad36865185b335cb7fac64997939291ae839620 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...')