Skip to content

Commit fa8a76d

Browse files
committed
add reranker score
1 parent 919a53b commit fa8a76d

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

scripts/add_reranker_score.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -134,6 +134,6 @@ class ModelArgs:
134134
data[i]['neg_scores'].append(float(scores[score_idx]))
135135
score_idx += 1
136136

137-
with open(score_args.output_dir, 'w') as f:
137+
with open(score_args.output_file, 'w') as f:
138138
for d in data:
139139
f.write(json.dumps(d) + '\n')

0 commit comments

Comments
 (0)