explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-07-14 18:22:27 +03:00
Code Issues Packages Projects Releases Wiki Activity

black

Browse Source
This commit is contained in:
kadarakos 2023-05-03 13:26:53 +00:00
parent db361db874
commit a5b9e63664
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
spacy/pipeline/span_finder.py
View File

@ -229,7 +229,7 @@ class SpanFinder(TrainablePipe):
offset = 0 offset = 0
for i, doc in enumerate(docs): for i, doc in enumerate(docs):
doc.spans[self.predicted_key] = [] doc.spans[self.predicted_key] = []
doc_scores = scores[offset:offset + len(doc)] doc_scores = scores[offset : offset + len(doc)]
for j in range(len(doc)): for j in range(len(doc)):
start_token_score = doc_scores[j] start_token_score = doc_scores[j]
# If token is a START then start scanning following tokens # If token is a START then start scanning following tokens
@ -238,11 +238,11 @@ 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:
span = doc[j:k + 1] span = doc[j : k + 1]
doc.spans[self.predicted_key].append(span) doc.spans[self.predicted_key].append(span)
def update( def update(