explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-11-04 09:57:26 +03:00
Code Issues Packages Projects Releases Wiki Activity

flaky test fix suggestion, hand set bias terms

Browse Source
This commit is contained in:
kadarakos 2023-06-01 16:44:55 +00:00
parent 56de1076a1
commit 658c4aee35
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
spacy/tests/pipeline/test_span_finder.py
View File

@ -201,6 +201,11 @@ def test_span_finder_suggester():
docs[1].spans[SPANS_KEY] = [docs[1][3:5]] docs[1].spans[SPANS_KEY] = [docs[1][3:5]]
span_finder = nlp.add_pipe("span_finder", config={"spans_key": SPANS_KEY}) span_finder = nlp.add_pipe("span_finder", config={"spans_key": SPANS_KEY})
nlp.initialize() nlp.initialize()
# Setting bias terms to 1000 to always make it predict something.
scorer_model = span_finder.model.get_ref("scorer")
bias = scorer_model.layers[0].get_param("b")
bias += 1000
scorer_model.layers[0].set_param("b", bias)
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")(