INNER CODE UNIT · Python
lines
austin-starks/Promptimizer · graph.py:40
lines = line1 + line2
labels = [l.get_label() for l in lines]
legend = ax1.legend(lines, labels, loc='upper left', fontsize=12, frameon=True, shadow=True)
# Set title
plt.title('Training and Validation Fitness over Generations', fontsize=16, fontweight='bold')
# Adjust layout and display
plt.tight_layout()
plt.show()
# Save the figure
plt.savefig('fitness_over_generations.png', dpi=300, bbox_inches='tight')