update i2i

calI2I2.py

@@ -29,7 +29,7 @@ if __name__=="__main__":
             user_item_dict[key] = 1
         else:
             user_item_dict[key] = user_item_dict[key]+1
-        if items[2] not in item_dict:
+        if vid not in item_dict:
             item_dict[vid] = 1
         else:
             item_dict[vid] = item_dict[vid]+1
@@ -87,6 +87,8 @@ if __name__=="__main__":
         pair_score = v
         if item1 in item_dict:
             item_score1 = item_dict[item1]
+            if item_score1<10:
+                continue
             i2i_pro = float(pair_score)/(float(item_score1)+5)
             if i2i_pro<0.000001:
                 continue
@@ -100,6 +102,8 @@ if __name__=="__main__":
                 rec_item_dict[item2] = rec_list1
         if item2 in item_dict:
             item_score2 = item_dict[item2]
+            if item_score2<10:
+                continue
             i2i_pro = float(pair_score)/(float(item_score2)+5)
             if i2i_pro<0.000001:
                 continue