修改冷启动队列

eda/smart_rank/new_tag3_cold_start.py

@@ -150,11 +150,7 @@ def get_es_diary(keywords, city_tag_id=-1, version=False):
                 "terms": {
                     "content_level": [6, 5, 4, 3.5, 3]
                 }
-            }],
-            "should":
-            should_list,
-            "minimum_should_match":
-            1
+            }]
         }
     }
     q["sort"] = sort_list
@@ -301,19 +297,22 @@ if __name__ == "__main__":
 
     redis_client = redis.StrictRedis.from_url("redis://:ReDis!GmTx*0aN9@172.16.40.173:6379")
 
+    """
     tractate_list = get_es_tractate(keywords)
     answer_list = get_es_answer(keywords)
     redis_client.rpush("doris:tag_v3:coldstart:tractate", *tractate_list)
     redis_client.rpush("doris:tag_v3:coldstart:answer", *answer_list)
+    """
 
-    all_city_tag_id = get_all_city_id()
-    for city_id in all_city_tag_id:
-        if city_id != -1:
-            diary_list = get_es_diary(keywords, city_id)
-            redis_client.rpush("doris:tag_v3:coldstart:diary:" + str(city_id), *diary_list)
+    #all_city_tag_id = get_all_city_id()
+    #for city_id in all_city_tag_id:
+    #    if city_id != -1:
+    #        diary_list = get_es_diary(keywords, city_id)
+    #        redis_client.rpush("doris:tag_v3:coldstart:diary:" + str(city_id), *diary_list)
 
     diary_list = get_es_diary(keywords)
-    redis_client.rpush("doris:tag_v3:coldstart:diary:-1", *diary_list)
+    print(diary_list)
+    #redis_client.rpush("doris:tag_v3:coldstart:diary:-1", *diary_list)
 
     # print("diary: " + str(len(diary_list)))
     # print("tractate: " + str(len(tractate_list)))