c3a6f53a6523a086752c07047c888a1019ba5773,recommenders/rating_prediction/base_rating_prediction.py,BaseRatingPrediction,__init__,#BaseRatingPrediction#Any#Any#Any#,7

Before Change

        self.train = ReadFile(train_file, self.space_type)
        self.train.main_information()

        if test_file != "":
            self.test = ReadFile(test_file, self.space_type)
            self.test.main_information()
            self.dataset_users = sorted(list(set(self.train.list_users + self.test.list_users)))
            self.dataset_items = sorted(list(set(self.train.list_items + self.test.list_items)))

        else:
            self.dataset_users = self.train.list_users
            self.dataset_items = self.train.list_items

        for item_id, item in enumerate(self.dataset_items):
            self.dict_item_id[item] = item_id

        for user_id, user in enumerate(self.dataset_users):

After Change

        for user_id, user in enumerate(self.train_users):
            self.map_users[user] = user_id

        for u, user in enumerate(self.train_users):
            for item in self.train_feedback[user]:
                self.matrix[u][self.map_items[item]] = self.train_feedback[user][item]

    def train_baselines(self):
        for i in xrange(10):
            self.compute_bi()
            self.compute_bu()

Instances