FilesExpand file tree

 main

/

plot.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

24 lines (21 loc) · 829 Bytes

 main

/

plot.py

Copy path

Top

File metadata and controls

• Code
• Blame

24 lines (21 loc) · 829 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

from pathlib import Path

import matplotlib.pyplot as plt

def trainTestPlot(plot, train_accu, test_accu, train_losses, test_losses, model_name):

if plot:

Path('plot/').mkdir(parents=True, exist_ok=True)

plot1 = plt.figure(1)

plt.plot(train_accu, '-o')

plt.plot(test_accu, '-o')

plt.xlabel('epoch')

plt.ylabel('accuracy')

plt.legend(['Train','Test'])

plt.title('Train vs Test Accuracy')

plt.savefig('plot/'+model_name+'_train_test_acc.png')

plot2 = plt.figure(2)

plt.plot(train_losses,'-o')

plt.plot(test_losses,'-o')

plt.xlabel('epoch')

plt.ylabel('losses')

plt.legend(['Train','Test'])

plt.title('Train vs Test Losses')

plt.savefig('plot/'+model_name+'_train_test_loss.png')