From 03fa8ab1cce6b383be650bbd86bd7845eb5b8716 Mon Sep 17 00:00:00 2001 From: Jakub Kliszko Date: Sat, 13 May 2023 14:48:48 +0200 Subject: [PATCH] feat: add figure saving --- lab5/code/main.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/lab5/code/main.py b/lab5/code/main.py index 95cc1b74..9c66ce6e 100644 --- a/lab5/code/main.py +++ b/lab5/code/main.py @@ -101,6 +101,7 @@ plt.plot(loss_values) plt.xlabel('Learning Step') plt.ylabel('Loss') plt.title('Loss Value') +plt.savefig('loss_value.png') plt.show() # Plot the accuracy on train set after each epoch @@ -108,6 +109,7 @@ plt.plot(train_acc_values) plt.xlabel('Epoch') plt.ylabel('Train Accuracy') plt.title('Accuracy on Train Set') +plt.savefig('train_accuracy.png') plt.show() # Plot the accuracy on validation set after each epoch @@ -115,4 +117,5 @@ plt.plot(val_acc_values) plt.xlabel('Epoch') plt.ylabel('Validation Accuracy') plt.title('Accuracy on Validation Set') +plt.savefig('validation_accuracy.png') plt.show()