소스 검색

update i2i

linfan 1 년 전
부모
커밋
dfdefc9145
1개의 변경된 파일5개의 추가작업 그리고 1개의 파일을 삭제
  1. 5 1
      calI2I2.py

+ 5 - 1
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