forked from amitab/Recommendare
-
Notifications
You must be signed in to change notification settings - Fork 0
/
hypertarget.py
33 lines (24 loc) · 933 Bytes
/
hypertarget.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
from pymongo import MongoClient
from operator import itemgetter, attrgetter, methodcaller
import config
from Recommendare import Recommendare
from movie_wrapper import MovieWrapper
class Hypertarget:
def __init__(self, db = None):
if db == None:
client = MongoClient(config.db_config['host'], config.db_config['port'])
self.db = client.hypertarget_ads
else:
self.db = db
self.recommender = Recommendare(self.db)
self.movie_wrapper = MovieWrapper(self.db)
def hypertarget(self, user_id, count):
movie_set = self.recommender.recommend(user_id, count)
movies = []
for movie in movie_set:
movies.append({
'predicted_rating': movie[1],
'movie_id': movie[0],
'details': self.movie_wrapper.get_movie_details(movie[0])
})
return movies