Skip to content

Commit

Permalink
Update PoseBusters plot
Browse files Browse the repository at this point in the history
  • Loading branch information
amorehead committed Dec 17, 2024
1 parent 124227b commit 1fd8ef4
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions notebooks/posebusters_method_interaction_analysis_plotting.py
Original file line number Diff line number Diff line change
Expand Up @@ -523,11 +523,11 @@ def histogram_to_vector(histogram, bins):

# plot the EMD values for each method
emd_values_df = pd.DataFrame(emd_values, columns=["Category", "Target", "EMD"])
emd_values_df.to_csv("dockgen_plif_emd_values.csv")
emd_values_df.to_csv("posebusters_benchmark_plif_emd_values.csv")

plt.figure(figsize=(10, 5))
sns.boxplot(data=emd_values_df, x="Category", y="EMD")
plt.xlabel("")
plt.ylabel("PLIF-EMD")
plt.savefig("dockgen_plif_emd_values.png")
plt.savefig("posebusters_benchmark_plif_emd_values.png")
plt.show()

0 comments on commit 1fd8ef4

Please sign in to comment.