linfan hai 1 ano
pai
achega
e1241a112e
Modificáronse 2 ficheiros con 3 adicións e 3 borrados
  1. 2 2
      calI2I2.py
  2. 1 1
      import_redist.py

+ 2 - 2
calI2I2.py

@@ -30,9 +30,9 @@ if __name__=="__main__":
         else:
             user_item_dict[key] = user_item_dict[key]+1
         if items[2] not in item_dict:
-            item_dict[items[2]] = 1
+            item_dict[vid] = 1
         else:
-            item_dict[items[2]] = item_dict[items[2]]+1
+            item_dict[vid] = item_dict[vid]+1
     f.close()
     #((user,item), score)
     #print(user_item_dict)

+ 1 - 1
import_redist.py

@@ -18,7 +18,7 @@ if __name__=="__main__":
              res_info = json.dumps(rec_item_list)
              
              f2.write(str(line[0])+"\t"+res_info+"\n")
-             key="sim_hot_"+line[0]
+             key="sim_hot_"+str(line[0])
              import_data_dict[key] = res_info
          redis_helper = RedisHelper()
          redis_helper.update_batch_setnx_key(import_data_dict, 60*60*24*7)