-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathrelate.py
36 lines (32 loc) · 1.08 KB
/
relate.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
import pickle
import time
start_time = time.time()
with open("dumps/user_vs_movie.pickle","rb") as user_movie_matrix_dump:
user_movie=pickle.load(user_movie_matrix_dump)
print len(user_movie)
movie_movie=[[0]*1682 for _ in range(1682)]
for i in range(0,len(user_movie)):
for j in range(0,len(user_movie[i])-1):
k=j+1
while k<=len(user_movie[i])-1:
movie_movie[j][k]+=user_movie[i][j]*user_movie[i][k]
k=k+1
movie_movie_dump = open("dumps/movie_movie.pickle","wb")
pickle.dump(movie_movie,movie_movie_dump)
movie_movie_dump.close()
'''
#print user_movie[0][0:10]
movie_movie=[[0]*10 for _ in range(10)]
#print movie_movie
for i in range(0,10):
for j in range(0,9):
k=j+1
while k<=9:
#if user_movie[i][j]>0 and user_movie[i][k]:
#print i,j,k,user_movie[i][j],user_movie[i][k]
movie_movie[j][k]=movie_movie[j][k]+(user_movie[i][j]*user_movie[i][k])
k=k+1
for l in movie_movie:
print l,"\n"
'''
print time.time()-start_time