Update plot labels and annotations for better readability and clarity.

This commit is contained in:
Paul Gauthier 2024-05-15 11:13:17 -07:00
parent 75fa061016
commit 05e34783c2

View file

@ -23,18 +23,18 @@ def plot_over_time(yaml_file):
rc("font", **{"family": "sans-serif", "sans-serif": ["Helvetica"], "size": 10}) rc("font", **{"family": "sans-serif", "sans-serif": ["Helvetica"], "size": 10})
fig, ax = plt.subplots(figsize=(6, 4)) fig, ax = plt.subplots(figsize=(10, 4))
ax.grid(axis="y", zorder=0, lw=0.2) ax.grid(axis="y", zorder=0, lw=0.2)
colors = ['red' if 'gpt-4' in model else 'green' if 'gpt-3.5' in model else 'blue' for model in models] colors = ['red' if 'gpt-4' in model else 'green' if 'gpt-3.5' in model else 'blue' for model in models]
ax.scatter(dates, pass_rates, c=colors, alpha=0.5) ax.scatter(dates, pass_rates, c=colors, alpha=0.5)
for i, model in enumerate(models): for i, model in enumerate(models):
ax.annotate(model, (dates[i], pass_rates[i]), fontsize=8, alpha=0.75, ax.annotate(model, (dates[i], pass_rates[i]), fontsize=12, alpha=0.75,
xytext=(5, 5), textcoords='offset points') xytext=(5, 5), textcoords='offset points')
ax.set_xlabel('Release Date') ax.set_xlabel('Model release date')
ax.set_ylabel('Pass Rate 2') ax.set_ylabel('Aider code editing benchmark,\npercent completed correctly')
ax.set_title('Model Performance Over Time') ax.set_title('LLM code editing skill by model release date')
plt.tight_layout() plt.tight_layout()
plt.savefig("tmp_over_time.png") plt.savefig("tmp_over_time.png")
imgcat(fig) imgcat(fig)