blob: 4d0fc03edf62b959ca701f22ac9e4cf466da77b5 (
plain)
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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
|
#!/usr/bin/python
import os
import sys
import pickle
prun = 0
runs = {}
labels = {}
users = pickle.load((open(sys.argv[3])))
recs = map(lambda x:0,users)
for line in open(sys.argv[1]):
line = line.split(',')
try:
run = int(line[2])
except:
continue
user = line[1]
if run not in runs:
runs[run] = 0
labels[run] = users.index(user) + 1
for line in open(sys.argv[2]):
line = line.split(',')
try:
run = int(line[2])
except:
continue
user = line[1]
rec = ' '.join(line[6].split('@')[0].split('_'))
if run != prun and prun > 0:
runs[prun] = recs.index(max(recs))+1
recs = map(lambda x:0,users)
recs[users.index(rec)] += 1
prun = run
t=0.0
tp=0.0
fp=0.0
fn=0.0
for (k,v) in runs.items():
print v,labels[k]
if v == 0:
fn += 1
elif v != labels[k]:
fp += 1
else:
tp += 1
t += 1
print runs[167],labels[167]
print tp,fp,fn
print("True positives: ",tp/t)
print("False positives: ",fp/t)
print("False negatives: ",fn/t)
|
