14d9448978c808b4b86e5b3429d7649485782568,entity2rec/ItemKNNSImilarity.py,ItemKNNSimilarity,__init__,#ItemKNNSimilarity#,16

Before Change

        self.model = self.learn_model_surprise()

        self.sim_matrix = self.model.compute_similarities()

        self.avg_s = np.mean(self.sim_matrix)

    def learn_model_surprise(self):

        file_path = os.path.expanduser(self.train)

After Change

        self.dataset = dataset

        data = tc.SFrame.read_csv("datasets/"+"%s/FM/" %self.dataset
                                  +"train.dat", delimiter=" ")

        if implicit:

            self.model = tc.item_similarity_recommender.create(data,
                        user_id="user_id", item_id="item_id")

        else:

            self.model = tc.item_similarity_recommender.create(data,
                        user_id="user_id", item_id="item_id", target="rating")
        
        similarities = self.model.get_similar_items()
        
        self.sim_matrix = {}

        for s in similarities:

            self.sim_matrix[s["item_id"], s["similar"]] = s["score"]

    def collab_similarities(self, item_1, item_2):

        try:

Instances