kuhyx/WUT_Computer_Science
1
0
Fork 0
mirror of https://github.com/kuhyx/WUT_Computer_Science.git synced 2026-07-04 19:43:03 +02:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: add figure saving

Browse Source
This commit is contained in:
Jakub Kliszko 2023-05-13 14:48:48 +02:00
parent fe23eadb1a
commit 03fa8ab1cc
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
lab5/code/main.py
View File

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