-
Notifications
You must be signed in to change notification settings - Fork 0
/
cal_auaac.py
30 lines (25 loc) · 848 Bytes
/
cal_auaac.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
import numpy as np
import csv
import glob
import os
result_csvs = sorted(glob.glob('./result/*.csv'))
if os.path.exists('AUAAC.csv') == False:
with open('AUAAC.csv', 'w') as f:
writer = csv.writer(f)
writer.writerow(['name','AUAAC'])
for csvfile in result_csvs:
current_file = []
with open(csvfile) as f:
reader = csv.reader(f)
for i, row in enumerate(reader):
current_file.append( row[1:] )
np_current_file = np.asarray( current_file, dtype=float )
performance = np_current_file[0,:]
ood = np_current_file[1,:]
score = 0
for i in range(101):
ood_dist = ood[i+1] - ood[i]
score += (performance[i] + performance[i+1]) * ood_dist / 2
with open('./AUAAC.csv', 'a') as f:
writer = csv.writer(f)
writer.writerow([ csvfile, score ])