This commit is contained in:
kadarakos 2023-05-03 13:33:18 +00:00
parent 82f6a813f0
commit 6fe7c66f36

View File

@ -231,7 +231,7 @@ class SpanFinder(TrainablePipe):
doc.spans[self.predicted_key] = [] doc.spans[self.predicted_key] = []
starts = [] starts = []
ends = [] ends = []
doc_scores = scores[offset:offset + len(doc)] doc_scores = scores[offset : offset + len(doc)]
for token, token_score in zip(doc, doc_scores): for token, token_score in zip(doc, doc_scores):
if token_score[0] >= self.threshold: if token_score[0] >= self.threshold: