modify

linucb/views/linucb.py

@@ -87,10 +87,10 @@ class LinUCB:
                 for top_score_index in np_score_dict[top_score]:
                     tag_id = str(tag_list[top_score_index], encoding="utf-8")
                     top_tag_dict[tag_id] = top_score
-                    if len(top_tag_dict) >= 3:
+                    if len(top_tag_dict) >= 20:
                         break
 
-                if len(top_tag_dict) >= 3:
+                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)))

search/utils/topic.py

@@ -309,7 +309,7 @@ class TopicUtils(object):
 
             q["query"]["function_score"] = query_function_score
             q["collapse"] = {
-                "field": "user_id"
+                "field": "tag_id"
             }
             q["_source"] = {
                 "includes": ["id", "pictorial_id", "offline_score", "user_id", "edit_tag_list"]