ADD file via upload
This commit is contained in:
parent
6da390e100
commit
4e8a10f1ba
|
@ -0,0 +1,25 @@
|
|||
import torch
|
||||
import datetime
|
||||
import os
|
||||
|
||||
def save_result(data, ylabel, args):
|
||||
data = {'base' :data}
|
||||
|
||||
path = './output'
|
||||
|
||||
file = '{}_{}.txt'.format(args.algorithm, datetime.datetime.now().strftime("%Y_%m_%d_%H_%M_%S"))
|
||||
|
||||
if not os.path.exists(path):
|
||||
os.makedirs(path)
|
||||
|
||||
with open(os.path.join(path,file), 'a') as f:
|
||||
for label in data:
|
||||
f.write(label)
|
||||
f.write(' ')
|
||||
for item in data[label]:
|
||||
item1 = str(item)
|
||||
f.write(item1)
|
||||
f.write(' ')
|
||||
f.write('\n')
|
||||
print('save finished')
|
||||
f.close()
|
Loading…
Reference in New Issue