Skip to content
Snippets Groups Projects
Commit a59c05b1 authored by Charly Lamothe's avatar Charly Lamothe
Browse files

Fix plot in compute_results.py and fix model_weights getter in model_raw_results

parent ec55d270
No related branches found
No related tags found
1 merge request!10Resolve "Visualization"
This commit is part of merge request !10. Comments created here will be created in the context of that merge request.
...@@ -25,7 +25,7 @@ class ModelRawResults(object): ...@@ -25,7 +25,7 @@ class ModelRawResults(object):
@property @property
def model_weights(self): def model_weights(self):
return self.model_weights return self._model_weights
@property @property
def training_time(self): def training_time(self):
......
...@@ -138,7 +138,6 @@ def extract_weights_across_seeds(models_dir, results_dir, experiment_id): ...@@ -138,7 +138,6 @@ def extract_weights_across_seeds(models_dir, results_dir, experiment_id):
extracted_forest_sizes = os.listdir(extracted_forest_sizes_root_path) extracted_forest_sizes = os.listdir(extracted_forest_sizes_root_path)
extracted_forest_sizes = [nb_tree for nb_tree in extracted_forest_sizes if not 'no_weights' in nb_tree ] extracted_forest_sizes = [nb_tree for nb_tree in extracted_forest_sizes if not 'no_weights' in nb_tree ]
extracted_forest_sizes.sort(key=int) extracted_forest_sizes.sort(key=int)
all_extracted_forest_sizes.append(list(map(int, extracted_forest_sizes)))
for extracted_forest_size in extracted_forest_sizes: for extracted_forest_size in extracted_forest_sizes:
# models/{experiment_id}/seeds/{seed}/extracted_forest_sizes/{extracted_forest_size} # models/{experiment_id}/seeds/{seed}/extracted_forest_sizes/{extracted_forest_size}
extracted_forest_size_path = extracted_forest_sizes_root_path + os.sep + extracted_forest_size extracted_forest_size_path = extracted_forest_sizes_root_path + os.sep + extracted_forest_size
...@@ -392,7 +391,7 @@ if __name__ == "__main__": ...@@ -392,7 +391,7 @@ if __name__ == "__main__":
omp_with_params_experiment_score_metric = extract_scores_across_seeds_and_extracted_forest_sizes( omp_with_params_experiment_score_metric = extract_scores_across_seeds_and_extracted_forest_sizes(
args.models_dir, args.results_dir, args.experiment_ids[2]) args.models_dir, args.results_dir, args.experiment_ids[2])
#omp_with_params_without_weights #omp_with_params_without_weights
logger.info('Loading omp_with_params experiment scores...') logger.info('Loading omp_with_params without 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_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( 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) args.models_dir, args.results_dir, args.experiment_ids[2], weights=False)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment