- Fix bug in base knn

caserec · May 27, 2017 · c688514 · c688514
1 parent b904b13
commit c688514
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/caserec/recommenders/rating_prediction/base_knn.py b/caserec/recommenders/rating_prediction/base_knn.py
@@ -62,6 +62,7 @@ def compute_bi(self):
 
         for item in self.train['items']:
             cont = 0
+            # self.bi.update({item: 0})
             for user in self.train['di'][item]:
                 self.bi[item] = self.bi.get(item, 0) + float(self.train['feedback'][user].get(item, 0)) - \
                                 self.train['mean_rates'] - self.bu.get(user, 0)

diff --git a/caserec/utils/read_file.py b/caserec/utils/read_file.py
@@ -121,7 +121,7 @@ def return_information(self, implicit=False):
                     matrix[map_user[user]][map_item[item]] = 1
                 else:
                     matrix[map_user[user]][map_item[item]] = d_feedback[user][item]
-                self.dict_items.setdefault(map_item[item], set()).add(map_user[user])
+                # self.dict_items.setdefault(map_item[item], set()).add(map_user[user])
 
         sparsity = (1 - (self.number_interactions / float(len(self.list_users) * len(self.list_items)))) * 100
         dict_file.update({'feedback': d_feedback, 'users': self.list_users, 'items': self.list_items,

diff --git a/setup.py b/setup.py
@@ -6,13 +6,13 @@
 setup(
     name='CaseRecommender',
     packages=find_packages(),
-    version='0.0.15',
+    version='0.0.16',
     license='GPL3 License',
     description='A recommender systems framework for Python',
     author='Arthur Fortes',
     author_email='fortes.arthur@gmail.com',
     url='https://github.com/ArthurFortes/CaseRecommender',
-    download_url='https://github.com/ArthurFortes/CaseRecommender/tarball/0.0.15',
+    download_url='https://github.com/ArthurFortes/CaseRecommender/tarball/0.0.16',
     keywords=['recommender systems', 'framework', 'collaborative filtering', 'content-based filtering'],
     classifiers=[],
 )