explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2024-12-25 09:26:27 +03:00
Code Issues Packages Projects Releases Wiki Activity

make EntityLinker robust for nO=None (#7930)

Browse Source
This commit is contained in:
Sofie Van Landeghem 2021-05-06 10:14:47 +02:00 committed by GitHub
parent 66bfabd839
commit e9037d8fc0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
spacy/ml/models/entity_linker.py
View File

@ -11,7 +11,7 @@ from ...vocab import Vocab
@registry.architectures("spacy.EntityLinker.v1") @registry.architectures("spacy.EntityLinker.v1")
def build_nel_encoder(tok2vec: Model, nO: Optional[int] = None) -> Model: def build_nel_encoder(tok2vec: Model, nO: Optional[int] = None) -> Model:
with Model.define_operators({">>": chain, "**": clone}): with Model.define_operators({">>": chain, "**": clone}):
token_width = tok2vec.get_dim("nO") token_width = tok2vec.maybe_get_dim("nO")
output_layer = Linear(nO=nO, nI=token_width) output_layer = Linear(nO=nO, nI=token_width)
model = ( model = (
tok2vec tok2vec