mirror of
https://github.com/explosion/spaCy.git
synced 2025-07-10 16:22:29 +03:00
use the 'spans_key' variable name everywhere
This commit is contained in:
parent
fe964e7831
commit
8c7c34d4f4
|
@ -321,7 +321,7 @@ class SpanFinder(TrainablePipe):
|
||||||
|
|
||||||
|
|
||||||
@registry.misc("spacy.span_finder_suggester.v1")
|
@registry.misc("spacy.span_finder_suggester.v1")
|
||||||
def build_span_finder_suggester(candidates_key: str) -> Suggester:
|
def build_span_finder_suggester(spans_key: str) -> Suggester:
|
||||||
"""Suggest every candidate predicted by the SpanFinder"""
|
"""Suggest every candidate predicted by the SpanFinder"""
|
||||||
|
|
||||||
def span_finder_suggester(
|
def span_finder_suggester(
|
||||||
|
@ -333,8 +333,8 @@ def build_span_finder_suggester(candidates_key: str) -> Suggester:
|
||||||
lengths = []
|
lengths = []
|
||||||
for doc in docs:
|
for doc in docs:
|
||||||
length = 0
|
length = 0
|
||||||
if doc.spans[candidates_key]:
|
if doc.spans[spans_key]:
|
||||||
for span in doc.spans[candidates_key]:
|
for span in doc.spans[spans_key]:
|
||||||
spans.append([span.start, span.end])
|
spans.append([span.start, span.end])
|
||||||
length += 1
|
length += 1
|
||||||
|
|
||||||
|
|
|
@ -204,7 +204,7 @@ def test_span_finder_suggester():
|
||||||
span_finder.set_annotations(docs, span_finder.predict(docs))
|
span_finder.set_annotations(docs, span_finder.predict(docs))
|
||||||
|
|
||||||
suggester = registry.misc.get("spacy.span_finder_suggester.v1")(
|
suggester = registry.misc.get("spacy.span_finder_suggester.v1")(
|
||||||
candidates_key=SPANS_KEY
|
spans_key=SPANS_KEY
|
||||||
)
|
)
|
||||||
|
|
||||||
candidates = suggester(docs)
|
candidates = suggester(docs)
|
||||||
|
|
Loading…
Reference in New Issue
Block a user