Skip to content
Snippets Groups Projects
Commit f5dfbf7f authored by Marie Generali's avatar Marie Generali :worried: Committed by Mélodie
Browse files

Add space confidence to the confidence by ner token

parent 619e8822
No related branches found
No related tags found
No related merge requests found
......@@ -353,8 +353,8 @@ def run(
result["confidences"]["by ner token"] = [
{
"text": f"{text[current: next_token-1]}",
"confidence_ner": f"{np.around(np.mean(char_confidences[current : next_token-1]), 2)}",
"text": f"{text[current: next_token]}",
"confidence_ner": f"{np.around(np.mean(char_confidences[current : next_token]), 2)}",
}
for current, next_token in pairwise(index + [0])
]
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment