Commit 3cd5355f authored by Jerome Wuerf's avatar Jerome Wuerf
Browse files

Merge branch 'fix-ranking-score' into 'main'

Fix ranking score

See merge request !21
parents 7141dbdc 8a078b7e
......@@ -125,12 +125,11 @@ class Reranking:
if stance_conclusion == stance_premise \
and premise['_id'] not in encountered_premise_ids:
trec_style_rows.append(
(topic_nrb,
conclusion['_source']['sentence_stance'],
f'{conclusion["_id"]},{premise["_id"]}',
(topic_nrb, conclusion['_source']['sentence_stance'],
f'{"__".join(conclusion["_id"].split("_"))},' +
f'{"__".join(premise["_id"].split("_"))}',
len(trec_style_rows),
f'{conclusion["final_reranking_score"]:.3f}',
self.run_name))
f'{conclusion["final_reranking_score"]:.2f}', self.run_name))
encountered_premise_ids.add(premise['_id'])
break
rows_per_topic[topic_nrb] = trec_style_rows
......
Supports Markdown
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