s495719/ium_495719
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Change plot

Browse Source
This commit is contained in:
PawelDopierala 2024-05-15 01:39:13 +02:00
parent 239bc6d81a
commit aa45e2b730
1 changed files with 10 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
evaluate.py
View File

@ -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()