Skip to content
Snippets Groups Projects

Correct research of predicted entity beginning

Merged Thibault Lavigne requested to merge correct_entities_beginning into master
2 files
+ 101
1
Compare changes
  • Side-by-side
  • Inline
Files
2
+ 11
1
@@ -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])
Loading