diff --git a/use_model.py b/use_model.py index 301d684..f1f2f02 100644 --- a/use_model.py +++ b/use_model.py @@ -34,6 +34,7 @@ def create_plots(): plt.xticks(range(min(metrics_df["Build number"]), max(metrics_df["Build number"]) + 1)) plt.show() plt.savefig("Accuracy_plot.png") + plt.close() plt.plot(metrics_df["Build number"], metrics_df["F1 Score"]) plt.xlabel("Build Number") @@ -42,6 +43,7 @@ def create_plots(): plt.xticks(range(min(metrics_df["Build number"]), max(metrics_df["Build number"]) + 1)) plt.show() plt.savefig("F1_score_plot.png") + plt.close() plt.plot(metrics_df["Build number"], metrics_df["RMSE"]) plt.xlabel("Build Number") @@ -50,6 +52,7 @@ def create_plots(): plt.xticks(range(min(metrics_df["Build number"]), max(metrics_df["Build number"]) + 1)) plt.show() plt.savefig("RMSE_plot.png") + plt.close() np.set_printoptions(threshold=20)