Skip to content

C
covid_analysis
Commit 04e7acb1 authored by Joaquin Torres's avatar Joaquin Torres
Browse files
Options

method names corrected for plots

parent e45e7474
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
model_selection/test_models.py
View file @ 04e7acb1
...@@ -277,7 +277,7 @@ if __name__ == "__main__": ...@@ -277,7 +277,7 @@ if __name__ == "__main__":
scores_df.at[model_name, score_name] = score_value scores_df.at[model_name, score_name] = score_value
# Adjust layout and save/show figure # Adjust layout and save/show figure
plt.tight_layout() plt.tight_layout()
plt.savefig(f'./test_results/roc_pr_curves/{group}_{method}.svg', format='svg', dpi=500) plt.savefig(f'./test_results/roc_pr_curves/{group}_{method_names[j]}.svg', format='svg', dpi=500)
plt.close(fig) plt.close(fig)
# Store the DataFrame in the dictionary with a unique key for each sheet # Store the DataFrame in the dictionary with a unique key for each sheet
sheet_name = f"{group}_{method_names[j]}" sheet_name = f"{group}_{method_names[j]}"
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment