Skip to content
Snippets Groups Projects
Commit de63b775 authored by Marie Generali's avatar Marie Generali :worried: Committed by Yoann Schneider
Browse files

Add space confidence to the confidence by ner token

parent f57bc667
No related branches found
No related tags found
1 merge request!170Add space confidence to the confidence by ner token
......@@ -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