Skip to content
Snippets Groups Projects
Commit b2db7c00 authored by Blanche Miret's avatar Blanche Miret
Browse files

Correct research of predicted entity beginning

parent a1613265
No related branches found
No related tags found
1 merge request!5Correct research of predicted entity beginning
Pipeline #103796 passed
......@@ -273,7 +273,17 @@ def compute_matches(
and j not in visited_predict
):
j -= 1
current_compar += prediction[j + 1 : i]
if (
"B" in labels_predict[j]
and get_type_label(labels_predict[j]) == tag_ref
and j not in visited_predict
):
start = j
else:
start = j + 1
current_compar += prediction[start:i]
found_aligned_beginning = True
current_compar.append(prediction[i])
......
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