Skip to content

P
physical
Commit 05e45ec8 authored by 段英荣's avatar 段英荣
Browse files
Options

modify

parent 6eae2072
Hide whitespace changes
Inline Side-by-side
Showing with 31 additions and 19 deletions
search/utils/topic.py
View file @ 05e45ec8
......@@ -134,7 +134,7 @@ class TopicUtils(object):
"""
try:
attention_user_id_list = list()
pick_user_id_list = list()
# pick_user_id_list = list()
# same_group_id_list = list()
user_tag_list = list()
......@@ -145,8 +145,8 @@ class TopicUtils(object):
attention_user_info_list = result_dict["hits"][0]["_source"]["attention_user_id_list"]
attention_user_id_list = [item["user_id"] for item in attention_user_info_list]
pick_user_info_list = result_dict["hits"][0]["_source"]["pick_user_id_list"]
pick_user_id_list = [item["user_id"] for item in pick_user_info_list]
# pick_user_info_list = result_dict["hits"][0]["_source"]["pick_user_id_list"]
# pick_user_id_list = [item["user_id"] for item in pick_user_info_list]
# same_pictorial_user_info_list = result_dict["hits"][0]["_source"]["same_pictorial_user_id_list"]
#
......@@ -165,7 +165,7 @@ class TopicUtils(object):
"language_type": 1
}
},
"weight": 3
"weight": 4
},
{
"gauss": {
......@@ -174,7 +174,7 @@ class TopicUtils(object):
"decay": 0.99
}
},
"weight": 3
"weight": 5
}
]
......@@ -197,14 +197,14 @@ class TopicUtils(object):
"weight": 3,
}
)
if len(pick_user_id_list) > 0:
functions_list.append(
{
"filter": {"bool": {
"should": {"terms": {"user_id": pick_user_id_list}}}},
"weight": 2
}
)
# if len(pick_user_id_list) > 0:
# functions_list.append(
# {
# "filter": {"bool": {
# "should": {"terms": {"user_id": pick_user_id_list}}}},
# "weight": 2
# }
# )
# if len(same_pictorial_id_list) > 0:
# functions_list.append(
......@@ -225,12 +225,24 @@ class TopicUtils(object):
}
)
if len(recommend_tag_list)>0:
functions_list.append(
{
"filter": {"terms": {"tag_list": recommend_tag_list}},
"weight": 3
}
)
if len(recommend_tag_list)>1:
functions_list += [
{
"filter": {"terms": {"tag_list": recommend_tag_list[0]}},
"weight": 4
},
{
"filter": {"terms": {"tag_list": recommend_tag_list[1:]}},
"weight": 3
}
]
else:
functions_list.append(
{
"filter": {"terms": {"tag_list": recommend_tag_list}},
"weight": 3
}
)
# for tag_id in recommend_tag_dict:
# functions_list.append(
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment