调试

linucb/core/Linucb.py

@@ -223,26 +223,12 @@ if __name__ == "__main__":
     #     [1, 2, 6, 4, 2, 1],
     # ]
 
-    user_features = [
-        ["1", "2", "9", "1", "3", "b"],
-        ["2", "2", "1", "2", "2", "1"],
-        ["3", "7", "1", "1", "3", "1"],
-        ["4", "2", "4", "5", "3", "a"],
-        ["3", "2", "r", "1", "4", "1"],
-        ["5", "5", "2", "3", "3", "1"],
-        ["3", "2", "1", "1", "3", "1"],
-        ["7", "3", "5", "1", "5", "1"],
-        ["8", "2", "1", "9", "3", "c"],
-        ["2", "2", "7", "1", "32", "1"],
-    ]
 
-    LinUCB.init_all_arm_by_card_content(user_features_list=user_features)
 
-
-    test_user_feature = ["1", "2", "9", "1", "3", "b"]
+    test_user_feature = [154.0,0.0,0.0]
 
     begin=time.time()
-    all_diary_content_redis_dict = redis_client.hgetall("strategy:linucb:content_type:diary")
+    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_user_feature,list(all_diary_content_redis_dict.keys()))
     print(time.time()-begin)
     print(top_tag_dict)
\ No newline at end of file