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

Back off to ORTH without Vectors.attr

Browse Source
This commit is contained in:
Adriane Boyd 2023-05-12 19:51:21 +02:00
parent b44f94dafa
commit 79bd698b73
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
spacy/ml/staticvectors.py
View File

@ -6,6 +6,7 @@ from thinc.util import partial
from thinc.types import Ragged, Floats2d, Floats1d, Ints1d from thinc.types import Ragged, Floats2d, Floats1d, Ints1d
from thinc.api import Model, Ops, registry from thinc.api import Model, Ops, registry
from ..attrs import ORTH
from ..tokens import Doc from ..tokens import Doc
from ..errors import Errors, Warnings from ..errors import Errors, Warnings
from ..vectors import Mode from ..vectors import Mode
@ -44,7 +45,9 @@ def forward(
if not token_count: if not token_count:
return _handle_empty(model.ops, model.get_dim("nO")) return _handle_empty(model.ops, model.get_dim("nO"))
vocab: Vocab = docs[0].vocab vocab: Vocab = docs[0].vocab
key_attr: int = vocab.vectors.attr key_attr: int = ORTH
if hasattr(vocab.vectors, "attr"):
key_attr = vocab.vectors.attr
keys = model.ops.flatten([cast(Ints1d, doc.to_array(key_attr)) for doc in docs]) keys = model.ops.flatten([cast(Ints1d, doc.to_array(key_attr)) for doc in docs])
W = cast(Floats2d, model.ops.as_contig(model.get_param("W"))) W = cast(Floats2d, model.ops.as_contig(model.get_param("W")))
if vocab.vectors.mode == Mode.default: if vocab.vectors.mode == Mode.default: