Skip to content
Snippets Groups Projects
Commit 8d911afd authored by Léo Bouscarrat's avatar Léo Bouscarrat
Browse files

Add log to find no_weight bug

parent 69f71671
No related branches found
No related tags found
2 merge requests!20Resolve "integration-sota",!19WIP: Resolve "Adding new datasets"
......@@ -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_])
......
......@@ -365,11 +365,13 @@ 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...')
_, _, base_with_params_test_scores_2, \
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment