This commit is contained in:
Paul Gauthier 2023-06-29 19:43:06 -07:00
parent a577a76a44
commit cf171bdd1e

View file

@ -65,10 +65,11 @@ def show_stats(dirnames):
df = pd.DataFrame.from_records(rows)
df.sort_values(by=["model", "edit_format"], inplace=True)
df_grouped = df.groupby(["model", "edit_format"])["pass_rate_1"].mean()
df_grouped1 = df.groupby(["model", "edit_format"])["pass_rate_1"].mean()
df_grouped2 = df.groupby(["model", "edit_format"])["pass_rate_2"].mean()
fig, ax = plt.subplots(figsize=(10, 6))
df_grouped.unstack().plot(kind="barh", ax=ax)
df_grouped2.unstack().plot(kind="barh", ax=ax)
ax.set_xlabel("Pass Rate 1")
ax.set_ylabel("Model")