Ejemplo n.º 1
0
 def __init__(self, data, metric=DEFAULT_METRIC, model_size=50):
     self.items = defaultdict(dict)
     for user, ratings in data.user_ratings.iteritems():
         for item, rating in ratings.iteritems():
             self.items[item][user] = rating
     Recommender.__init__(self, self.items, metric)
     self.data = data
     self.model_size = model_size
     
     self.item_similars = {}
     if exists(data.item_based_model_path):
         self.load_model(data.item_based_model_path)
     else:
         self.learn_model(data.item_based_model_path)
Ejemplo n.º 2
0
 def __init__(self):
     Recommender.__init__(self)
     self.weightSim = weightSim
     self.nNeigh = nNeigh