Skip to content

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

change test file

parent a8a84587
Hide whitespace changes
Inline Side-by-side
Showing with 25 additions and 31 deletions
rerank_esmm.py
View file @ ecb3e100
......@@ -87,43 +87,37 @@ def tag_boost(cid_str, tag_list=[15,21,22,85,86]):
cursor.execute(sql)
result = cursor.fetchall()
if len(result) > 0:
tmp_dict = {}
tag_cids = {}
left_cids = []
for i in result:
tmp = i[1].split(",")
tmp_dict[i[0]] = tmp
tag_cids[i[0]] = tmp
left_cids.extend(tmp)
tag_cids = {}
for i in tag_list:
if i in tmp_dict.keys():
tag_cids[i] = tmp_dict[i]
print(tag_list)
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 = []
n = 0
while n == len(tag_cids)-1:
for i in tag_list:
if i in tag_cids.keys():
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
if len(tag_cids["right"]) > 0:
sort_cids.extend(tag_cids["right"])
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
print(sort_cids)
print(len(sort_cids))
print(len(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