Skip to content

F
ffm-baseline
Commit 5e761f5f authored by 张彦钊's avatar 张彦钊
Browse files
Options

change test file

parent 51e53cac
Hide whitespace changes
Inline Side-by-side
Showing with 31 additions and 25 deletions
rerank_esmm.py
View file @ 5e761f5f
...@@ -87,36 +87,42 @@ def tag_boost(cid_str, tag_list=[15,21,22,85,86]): ...@@ -87,36 +87,42 @@ def tag_boost(cid_str, tag_list=[15,21,22,85,86]):
cursor.execute(sql) cursor.execute(sql)
result = cursor.fetchall() result = cursor.fetchall()
if len(result) > 0: if len(result) > 0:
tag_cids = {} tmp_dict = {}
left_cids = [] left_cids = []
for i in result: for i in result:
tmp = i[1].split(",") tmp = i[1].split(",")
tag_cids[i[0]] = tmp tmp_dict[i[0]] = tmp
left_cids.extend(tmp) left_cids.extend(tmp)
left_cids = list(set(left_cids)) tag_cids = {}
right_cids = [i for i in cids if i not in left_cids] for i in tag_list:
tag_cids["right"] = right_cids if i in tmp_dict.keys():
tag_list.append("right") tag_cids[i] = tmp_dict[i]
sort_cids = [] print(tag_cids)
while True:
for i in tag_list: # left_cids = list(set(left_cids))
if i in tag_cids.keys(): # right_cids = [i for i in cids if i not in left_cids]
n = 0 # tag_cids["right"] = right_cids
if len(tag_cids[i]) > 0: # tag_list.append("right")
sort_cids.append(tag_cids[i][0]) # sort_cids = []
value = tag_cids[i] # while True:
value.pop(0) # for i in tag_list:
tag_cids[i] = value # if i in tag_cids.keys():
if len(value) == 0 and i != "right": # n = 0
n = n + 1 # if len(tag_cids[i]) > 0:
else: # sort_cids.append(tag_cids[i][0])
n = n +1 # value = tag_cids[i]
if n == len(tag_cids)-1: # value.pop(0)
if len(tag_cids["right"]) > 0: # tag_cids[i] = value
sort_cids.extend(tag_cids["right"]) # if len(value) == 0 and i != "right":
print(sort_cids) # n = n + 1
return sort_cids # else:
# n = n +1
# if n == len(tag_cids)-1:
# if len(tag_cids["right"]) > 0:
# sort_cids.extend(tag_cids["right"])
# print(sort_cids)
# return sort_cids
else: else:
......
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