plt.close() fixup

This commit is contained in:
Starbeamrainbowlabs 2022-10-21 15:23:54 +01:00
parent 12dad3bc87
commit 42aea7a0cc
Signed by: sbrl
GPG key ID: 1BE5172E637709C2

View file

@ -26,3 +26,4 @@ def segmentation_plot(water_actual, water_predict, model_code, filepath_output):
plt.suptitle(f"Rainfall → Water depth prediction | {model_code}", fontsize=28, weight="bold") plt.suptitle(f"Rainfall → Water depth prediction | {model_code}", fontsize=28, weight="bold")
plt.savefig(filepath_output) plt.savefig(filepath_output)
plt.close()