Skip to content
GitLab
Menu
Projects
Groups
Snippets
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
Luc Giffon
bolsonaro
Commits
a59c05b1
Commit
a59c05b1
authored
Mar 06, 2020
by
Charly Lamothe
Browse files
Fix plot in compute_results.py and fix model_weights getter in model_raw_results
parent
ec55d270
Changes
2
Hide whitespace changes
Inline
Side-by-side
code/bolsonaro/models/model_raw_results.py
View file @
a59c05b1
...
...
@@ -25,7 +25,7 @@ class ModelRawResults(object):
@
property
def
model_weights
(
self
):
return
self
.
model_weights
return
self
.
_
model_weights
@
property
def
training_time
(
self
):
...
...
code/compute_results.py
View file @
a59c05b1
...
...
@@ -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
=
[
nb_tree
for
nb_tree
in
extracted_forest_sizes
if
not
'no_weights'
in
nb_tree
]
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
:
# 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
...
...
@@ -392,7 +391,7 @@ 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_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_experiment_score_metric
=
extract_scores_across_seeds_and_extracted_forest_sizes
(
args
.
models_dir
,
args
.
results_dir
,
args
.
experiment_ids
[
2
],
weights
=
False
)
...
...
Write
Preview
Supports
Markdown
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment