调试

linucb/core/Linucb.py

@@ -22,7 +22,7 @@ import time
 
 class LinUCB:
     d = 2
-    alpha = 0.1
+    alpha = 10
     r1 = 10
     r0 = -0.1
     default_tag_list = list()

tools/generate_content_user_feature.py

@@ -258,7 +258,16 @@ if __name__ == "__main__":
     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]]
+    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")