-
Notifications
You must be signed in to change notification settings - Fork 9
/
Copy pathvis_normalize_scores.py
38 lines (30 loc) · 1.15 KB
/
vis_normalize_scores.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
import sys
import os
import glob
if __name__=="__main__":
if len(sys.argv)<2:
print "Usage python vis_normalize_scores.py <input-folder>"
else:
dir_src = sys.argv[1]
csvlist = glob.glob(os.path.join(dir_src,"*.csv"))
global_max = -1000.0
global_min = 1000.0
for csvfile in csvlist:
print csvfile
with open(csvfile,'r') as f:
for line in f:
csvscore = float(line.strip().split()[1])
if csvscore > global_max:
global_max = csvscore
elif csvscore < global_min:
global_min = csvscore
for csvfile in csvlist:
print csvfile
of = open(csvfile.replace(".csv","_normalized.csv"),'w')
with open(csvfile, 'r') as f:
for line in f:
fname = line.strip().split(' ')[0]
csvscore = float(line.strip().split(' ')[1])
norm_csv_score = 10*(csvscore-global_min)/global_max
of.write(fname+','+str(norm_csv_score)+'\n')
of.close()