diff --git a/evaluate.py b/evaluate.py index aa6b0fe..4770716 100644 --- a/evaluate.py +++ b/evaluate.py @@ -39,17 +39,13 @@ else: updated_metrics_df.to_csv(metrics_file, index=False) -plt.figure(figsize=(10, 6)) -plt.plot(updated_metrics_df['Build_Number'], updated_metrics_df['RMSE'], label='RMSE', marker='o') -plt.plot(updated_metrics_df['Build_Number'], updated_metrics_df['MAE'], label='MAE', marker='o') -plt.plot(updated_metrics_df['Build_Number'], updated_metrics_df['R2'], label='R2', marker='o') - -plt.title('Metrics vs Builds') -plt.xlabel('Build Number') -plt.ylabel('Metric Value') -plt.legend() -plt.grid(True) - -plot_file = 'metrics_plt.png' -plt.savefig(plot_file) -plt.close() +metrics = ['RMSE', 'MAE', 'R2'] +for metric in metrics: + plt.plot(updated_metrics_df['Build_Number'], updated_metrics_df[metric], marker='o') + plt.title(f'{metric} vs Builds') + plt.xlabel('Build Number') + plt.ylabel(metric) + plt.grid(True) + plot_file = f'{metric.lower()}_plt.png' + plt.savefig(plot_file) + plt.close()