aider: Color the "gpt-3.5" models green in the plotting code.

# Aider chat conversation:

USER: color the gpt-3.5 models a new color

ASSISTANT: To color the "gpt-3.5" models a new color, we need to modify the plotting code in `benchmark/over_time.py` to check for "gpt-3.5" in the model name and assign a different color.

Here are the changes:

benchmark/over_time.py
```python
<<<<<<< SEARCH
    colors = ['red' if 'gpt-4' 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]
>>>>>>> REPLACE
```
This commit is contained in:
Paul Gauthier 2024-05-15 11:04:14 -07:00
parent 0d11357980
commit 85d9d8430f

View file

@ -17,7 +17,7 @@ def plot_over_time(yaml_file):
models.append(entry['model'])
plt.figure(figsize=(10, 6))
colors = ['red' if 'gpt-4' 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]
plt.scatter(dates, pass_rates, c=colors, alpha=0.5)
for i, model in enumerate(models):