yujunjun/FlagEmbedding
1
0
Fork 0
mirror of https://github.com/FlagOpen/FlagEmbedding.git synced 2026-01-01 17:51:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

add reranker score

Browse Source
This commit is contained in:
cfli 2024-10-29 20:27:36 +08:00
parent 919a53bcb6
commit fa8a76d7b7
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
scripts/add_reranker_score.py
View File

@ -134,6 +134,6 @@ if __name__ == '__main__':
data[i]['neg_scores'].append(float(scores[score_idx]))
score_idx += 1
with open(score_args.output_dir, 'w') as f:
with open(score_args.output_file, 'w') as f:
for d in data:
f.write(json.dumps(d) + '\n')