/
test.py
65 lines (50 loc) · 1.85 KB
/
test.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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
from algorithms import map
import csv
kmeans_predicted_results_file = "data/results/kmeans_predict.csv"
kmeans_test_results_file = "data/results/kmeans_test.csv"
user_cf_predict_results_file = "data/results/user_cf_predict.csv"
user_cf_mr_predict_results_file = "data/results/user_cf_mr_predict.csv"
user_cf_mr_test_results_file = "data/results/user_cf_mr_test.csv"
if __name__ == '__main__':
kmeans_test_results = []
kmeans_predicted_results = []
user_cf_predict_results = []
user_cf_test_results = []
user_cf_mr_predict_results = []
user_cf_mr_test_results = []
# f = open(kmeans_predicted_results_file, 'rt')
# reader = csv.reader(f)
# for row in reader:
# user = row[0]
# songs = row[1:]
# kmeans_predicted_results.append(songs)
# f = open(kmeans_test_results_file, 'rt')
# reader = csv.reader(f)
# for row in reader:
# user = row[0]
# songs = row[1:]
# kmeans_test_results.append(songs)
f = open(user_cf_mr_test_results_file, 'rt')
reader = csv.reader(f)
for row in reader:
user = row[0]
songs = row[1:]
user_cf_mr_test_results.append(songs)
f = open(user_cf_mr_predict_results_file, 'rt')
reader = csv.reader(f)
for row in reader:
user = row[0]
songs = row[1:]
user_cf_mr_predict_results.append(songs)
f = open(user_cf_predict_results_file, 'rt')
reader = csv.reader(f)
for row in reader:
user = row[0]
songs = row[1:]
user_cf_predict_results.append(songs)
# print "kmeans"
# print map.kdd_mapk(kmeans_test_results, kmeans_predicted_results, 500)
print "user_cf_mr"
print map.kdd_mapk(user_cf_mr_test_results, user_cf_mr_predict_results,500)
print "user_cf"
print map.kdd_mapk(user_cf_mr_test_results, user_cf_predict_results,500)