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]):
cursor.execute(sql)
result = cursor.fetchall()
if len(result) > 0:
tag_cids = {}
tmp_dict = {}
left_cids = []
for i in result:
tmp = i[1].split(",")
tag_cids[i[0]] = tmp
tmp_dict[i[0]] = tmp
left_cids.extend(tmp)
left_cids = list(set(left_cids))
right_cids = [i for i in cids if i not in left_cids]
tag_cids["right"] = right_cids
tag_list.append("right")
sort_cids = []
while True:
for i in tag_list:
if i in tag_cids.keys():
n = 0
if len(tag_cids[i]) > 0:
sort_cids.append(tag_cids[i][0])
value = tag_cids[i]
value.pop(0)
tag_cids[i] = value
if len(value) == 0 and i != "right":
n = n + 1
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
tag_cids = {}
for i in tag_list:
if i in tmp_dict.keys():
tag_cids[i] = tmp_dict[i]
print(tag_cids)
# left_cids = list(set(left_cids))
# right_cids = [i for i in cids if i not in left_cids]
# tag_cids["right"] = right_cids
# tag_list.append("right")
# sort_cids = []
# while True:
# for i in tag_list:
# if i in tag_cids.keys():
# n = 0
# if len(tag_cids[i]) > 0:
# sort_cids.append(tag_cids[i][0])
# value = tag_cids[i]
# value.pop(0)
# tag_cids[i] = value
# if len(value) == 0 and i != "right":
# n = n + 1
# 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:
......
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