This commit is contained in:
kadarakos 2023-05-03 13:26:53 +00:00
parent db361db874
commit a5b9e63664

View File

@ -238,7 +238,7 @@ class SpanFinder(TrainablePipe):
end_token_score = doc_scores[k] end_token_score = doc_scores[k]
# If token is an END check whether the lenght contraint is met # If token is an END check whether the lenght contraint is met
if end_token_score[1] >= self.threshold: if end_token_score[1] >= self.threshold:
span_length = (k + 1 - j) span_length = k + 1 - j
if span_length > self.max_length: if span_length > self.max_length:
break break
elif self.min_length <= span_length: elif self.min_length <= span_length: