mirror of
https://github.com/explosion/spaCy.git
synced 2025-07-04 03:43:09 +03:00
black
This commit is contained in:
parent
21d5360e67
commit
9cfcbdd0ad
|
@ -356,6 +356,7 @@ class SpanFinder(TrainablePipe):
|
||||||
else:
|
else:
|
||||||
self.model.initialize()
|
self.model.initialize()
|
||||||
|
|
||||||
|
|
||||||
@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(candidates_key: str) -> Suggester:
|
||||||
"""Suggest every candidate predicted by the SpanFinder"""
|
"""Suggest every candidate predicted by the SpanFinder"""
|
||||||
|
|
|
@ -62,9 +62,7 @@ def test_loss_alignment_example(tokens_predicted, tokens_reference, reference_tr
|
||||||
)
|
)
|
||||||
example = Example(predicted, reference)
|
example = Example(predicted, reference)
|
||||||
example.reference.spans[TRAINING_KEY] = [example.reference.char_span(5, 9)]
|
example.reference.spans[TRAINING_KEY] = [example.reference.char_span(5, 9)]
|
||||||
span_finder = nlp.add_pipe(
|
span_finder = nlp.add_pipe("span_finder", config={"training_key": TRAINING_KEY})
|
||||||
"span_finder", config={"training_key": TRAINING_KEY}
|
|
||||||
)
|
|
||||||
nlp.initialize()
|
nlp.initialize()
|
||||||
|
|
||||||
truth_scores = span_finder._get_aligned_truth_scores([example])
|
truth_scores = span_finder._get_aligned_truth_scores([example])
|
||||||
|
@ -100,9 +98,7 @@ def test_span_finder_component():
|
||||||
docs[0].spans[TRAINING_KEY] = [docs[0][3:4]]
|
docs[0].spans[TRAINING_KEY] = [docs[0][3:4]]
|
||||||
docs[1].spans[TRAINING_KEY] = [docs[1][3:5]]
|
docs[1].spans[TRAINING_KEY] = [docs[1][3:5]]
|
||||||
|
|
||||||
span_finder = nlp.add_pipe(
|
span_finder = nlp.add_pipe("span_finder", config={"training_key": TRAINING_KEY})
|
||||||
"span_finder", config={"training_key": TRAINING_KEY}
|
|
||||||
)
|
|
||||||
nlp.initialize()
|
nlp.initialize()
|
||||||
docs = list(span_finder.pipe(docs))
|
docs = list(span_finder.pipe(docs))
|
||||||
|
|
||||||
|
@ -159,9 +155,7 @@ def test_span_finder_suggester():
|
||||||
docs = [nlp("This is an example."), nlp("This is the second example.")]
|
docs = [nlp("This is an example."), nlp("This is the second example.")]
|
||||||
docs[0].spans[TRAINING_KEY] = [docs[0][3:4]]
|
docs[0].spans[TRAINING_KEY] = [docs[0][3:4]]
|
||||||
docs[1].spans[TRAINING_KEY] = [docs[1][3:5]]
|
docs[1].spans[TRAINING_KEY] = [docs[1][3:5]]
|
||||||
span_finder = nlp.add_pipe(
|
span_finder = nlp.add_pipe("span_finder", config={"training_key": TRAINING_KEY})
|
||||||
"span_finder", config={"training_key": TRAINING_KEY}
|
|
||||||
)
|
|
||||||
nlp.initialize()
|
nlp.initialize()
|
||||||
span_finder.set_annotations(docs, span_finder.predict(docs))
|
span_finder.set_annotations(docs, span_finder.predict(docs))
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue
Block a user