调试

linucb/core/Linucb.py

@@ -33,7 +33,7 @@ class LinUCB:
 
     redis_name_linucb_matrix_prefix = "strategy:linucb:content_type:"
     @classmethod
-    def linucb_recommend_tag(cls,device_id,redis_linucb_tag_data_dict,user_features_list,tag_list):
+    def linucb_recommend_tag(cls,redis_linucb_tag_data_dict,user_features_list,tag_list):
         """
         :remark 获取推荐标签
         :param redis_linucb_tag_data_dict:
@@ -96,7 +96,6 @@ class LinUCB:
                 if len(top_tag_dict) >= 20:
                     break
 
-            logging.info("duan add,device_id:%s,sorted_np_score_list:%s,np_score_dict:%s" % (str(device_id), str(sorted_np_score_list), str(np_score_dict)))
             return (top_tag_dict,top_tag_set)
         except:
             logging.error("catch exception,err_msg:%s" % traceback.format_exc())
@@ -220,4 +219,11 @@ if __name__ == "__main__":
         [1, 2, 6, 4, 2, 1],
     ]
 
-    LinUCB.init_all_arm_by_card_content(user_features_list=user_features)
\ No newline at end of file
+    LinUCB.init_all_arm_by_card_content(user_features_list=user_features)
+
+
+    test_user_feature = [1, 2, 1, 1, 3, 1]
+
+    all_diary_content_redis_dict = redis_client.hgetall("strategy:linucb: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(top_tag_dict)
\ No newline at end of file