调试

linucb/core/Linucb.py

@@ -96,10 +96,10 @@ class LinUCB:
                     tag_id = tag_list[top_score_index]
                     top_tag_dict[tag_id] = top_score
                     top_tag_set.add(tag_id)
-                    if len(top_tag_dict) >= 100:
+                    if len(top_tag_dict) >= 1:
                         break
 
-                if len(top_tag_dict) >= 100:
+                if len(top_tag_dict) >= 1:
                     break
 
             return (top_tag_dict,top_tag_set)

tools/generate_content_user_feature.py

@@ -223,7 +223,7 @@ if __name__ == "__main__":
     # Generate_Feature_Info.generate_content_feature_to_redis(label_encoder,onehot_encoder)
 
 
-    # linucb_matrix_redis_name = "strategy:linucb:matrix:content_type:diary"
+    linucb_matrix_redis_name = "strategy:linucb:matrix:content_type:diary"
     #
     # diary_click_file = "/data/log/duan_test/feed_query_data/feed_click_info.txt"
     # diary_fd = open(diary_click_file,"r")
@@ -257,23 +257,35 @@ if __name__ == "__main__":
     duan_device_id="868771031984211"
     user_feature_list = Generate_Feature_Info.get_user_feature_by_device_id(duan_device_id)
 
-    begin=time.time()
-    test_feature = [user_feature_list[0],user_feature_list[0]]
-    print(test_feature)
-    all_diary_content_redis_dict = redis_client.hgetall("strategy:linucb:matrix:content_type:diary")
-    top_tag_dict, top_tag_set = LinUCB.linucb_recommend_tag(all_diary_content_redis_dict,test_feature,list(all_diary_content_redis_dict.keys()))
-    print(time.time()-begin)
-    print(top_tag_dict)
-
-    print("*"*50)
-
-    begin=time.time()
-    test_feature = [1,2]
-    print(test_feature)
-    all_diary_content_redis_dict = redis_client.hgetall("strategy:linucb:matrix:content_type:diary")
-    top_tag_dict, top_tag_set = LinUCB.linucb_recommend_tag(all_diary_content_redis_dict,test_feature,list(all_diary_content_redis_dict.keys()))
-    print(time.time()-begin)
-    print(top_tag_dict)
+    test_count = 50
+    while test_count:
+        begin=time.time()
+        test_feature = [user_feature_list[0],user_feature_list[0]]
+        print(test_feature)
+        all_diary_content_redis_dict = redis_client.hgetall("strategy:linucb:matrix:content_type:diary")
+        top_tag_dict, top_tag_set = LinUCB.linucb_recommend_tag(all_diary_content_redis_dict,test_feature,list(all_diary_content_redis_dict.keys()))
+        print(time.time()-begin)
+        print(top_tag_dict)
+
+        for diary_id in top_tag_set:
+            update_feature_list = [user_feature_list[0]]
+            content_feature_list = Generate_Feature_Info.get_content_feature(diary_id)
+            if len(content_feature_list)>0:
+                update_feature_list.append(content_feature_list[0])
+            print(update_feature_list)
+            LinUCB.update_linucb_info(user_features=update_feature_list,reward=1,content_id=diary_id,redis_name_linucb_matrix=linucb_matrix_redis_name,redis_cli=redis_client)
+
+        test_count -= 1
+        print("*"*50)
+        time.sleep(2)
+
+    # begin=time.time()
+    # test_feature = [1,2]
+    # print(test_feature)
+    # all_diary_content_redis_dict = redis_client.hgetall("strategy:linucb:matrix:content_type:diary")
+    # top_tag_dict, top_tag_set = LinUCB.linucb_recommend_tag(all_diary_content_redis_dict,test_feature,list(all_diary_content_redis_dict.keys()))
+    # print(time.time()-begin)
+    # print(top_tag_dict)
 
     # test_val_list = ["切开双眼皮"]
     #