-
Notifications
You must be signed in to change notification settings - Fork 1
/
draw_and_save.py
39 lines (36 loc) · 1.52 KB
/
draw_and_save.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
import matplotlib.pyplot as plt
import os
import pandas as pd
import numpy as np
def curve_draw(train_loss_list, train_accurate_list, test_accurate_list, log_dir):
x1 = range(0, len(train_loss_list))
x2 = range(0, len(train_accurate_list))
x3 = range(0, len(test_accurate_list))
y1 = train_loss_list
y2 = train_accurate_list
y3 = test_accurate_list
plt.subplot(3, 1, 1)
plt.plot(x1, y1, 'o-')
plt.xlabel('epochs')
plt.subplot(3, 1, 2)
plt.plot(x2, y2, '*-')
plt.xlabel('epochs')
plt.subplot(3, 1, 3)
plt.plot(x3, y3, 'x-')
plt.xlabel('epochs')
#plt.savefig(log_dir + 'result' + version + '.png')
train_loss_list_path = log_dir + "train_loss" + '.csv'
train_accurate_list_path = log_dir + "train_accurate" + '.csv'
test_accurate_list_path = log_dir + "test_accurate" + '.csv'
train_loss_list_csv = pd.DataFrame(train_loss_list)
train_accurate_list_csv = pd.DataFrame(train_accurate_list)
test_accurate_list_csv = pd.DataFrame(test_accurate_list)
#with mox.file.File(train_loss_list_path,"w") as c1:
#train_loss_list_csv.to_csv(c1)
#with mox.file.File(train_accurate_list_path,"w") as c2:
#train_accurate_list_csv.to_csv(c2)
#with mox.file.File(test_accurate_list_path,"w") as c3:
#test_accurate_list_csv.to_csv(c3)
train_loss_list_csv.to_csv(train_loss_list_path)
train_accurate_list_csv.to_csv(train_accurate_list_path)
test_accurate_list_csv.to_csv(test_accurate_list_path)