Skip to content
Snippets Groups Projects
Commit 43c18761 authored by Luc Giffon's avatar Luc Giffon
Browse files

update results processing scripts

parent b1d518e5
No related branches found
1 merge request!24Resolve "non negative omp"
...@@ -14,10 +14,12 @@ lst_task_train_dev = ["coherence", "correlation"] ...@@ -14,10 +14,12 @@ lst_task_train_dev = ["coherence", "correlation"]
tasks = [ tasks = [
# "train_score", # "train_score",
# "dev_score", # "dev_score",
# "test_score", "test_score",
"coherence", # "coherence",
"correlation", # "correlation",
# "negative-percentage" # "negative-percentage",
"dev_strength",
"test_strength"
] ]
dct_score_metric_fancy = { dct_score_metric_fancy = {
...@@ -33,7 +35,7 @@ dct_color_by_strategy = { ...@@ -33,7 +35,7 @@ dct_color_by_strategy = {
"OMP Distillation w/o weights": (255, 128, 0), # orange "OMP Distillation w/o weights": (255, 128, 0), # orange
"OMP w/o weights": (255, 128, 0), # orange "OMP w/o weights": (255, 128, 0), # orange
"Random": (0, 0, 0), # black "Random": (0, 0, 0), # black
"Zhang Similarities": (255, 255, 0), # jaune "Zhang Similarities": (255,105,180), # rose
'Zhang Predictions': (128, 0, 128), # turquoise 'Zhang Predictions': (128, 0, 128), # turquoise
'Ensemble': (0, 0, 255), # blue 'Ensemble': (0, 0, 255), # blue
"Kmeans": (0, 255, 0) # red "Kmeans": (0, 255, 0) # red
...@@ -81,7 +83,7 @@ tpl_transparency = (0.1,) ...@@ -81,7 +83,7 @@ tpl_transparency = (0.1,)
if __name__ == "__main__": if __name__ == "__main__":
load_dotenv(find_dotenv('.env')) load_dotenv(find_dotenv('.env'))
dir_name = "bolsonaro_models_25-03-20" dir_name = "bolsonaro_models_27-03-20_v2"
dir_path = Path(os.environ["project_dir"]) / "results" / dir_name dir_path = Path(os.environ["project_dir"]) / "results" / dir_name
out_dir = Path(os.environ["project_dir"]) / "reports/figures" / dir_name out_dir = Path(os.environ["project_dir"]) / "reports/figures" / dir_name
...@@ -134,7 +136,7 @@ if __name__ == "__main__": ...@@ -134,7 +136,7 @@ if __name__ == "__main__":
title = "{} {}".format(task, data_name) title = "{} {}".format(task, data_name)
yaxis_title = "% negative weights" if task == "negative-percentage" else dct_score_metric_fancy[score_metric_name] yaxis_title = "% negative weights" if task == "negative-percentage" else dct_score_metric_fancy[score_metric_name]
fig.update_layout(barmode='group', fig.update_layout(barmode='group',
title=title, # title=title,
xaxis_title="# Selected Trees", xaxis_title="# Selected Trees",
yaxis_title=yaxis_title, yaxis_title=yaxis_title,
font=dict( font=dict(
......
...@@ -64,7 +64,8 @@ set_no_corr = set() ...@@ -64,7 +64,8 @@ set_no_corr = set()
if __name__ == "__main__": if __name__ == "__main__":
load_dotenv(find_dotenv('.env')) load_dotenv(find_dotenv('.env'))
dir_name = "results/bolsonaro_models_25-03-20" # dir_name = "results/bolsonaro_models_25-03-20"
dir_name = "results/bolsonaro_models_27-03-20_v2"
dir_path = Path(os.environ["project_dir"]) / dir_name dir_path = Path(os.environ["project_dir"]) / dir_name
output_dir_file = dir_path / "results.csv" output_dir_file = dir_path / "results.csv"
......
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