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( ...@@ -353,8 +353,8 @@ def run(
result["confidences"]["by ner token"] = [ result["confidences"]["by ner token"] = [
{ {
"text": f"{text[current: next_token-1]}", "text": f"{text[current: next_token]}",
"confidence_ner": f"{np.around(np.mean(char_confidences[current : next_token-1]), 2)}", "confidence_ner": f"{np.around(np.mean(char_confidences[current : next_token]), 2)}",
} }
for current, next_token in pairwise(index + [0]) 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