mirror of
https://github.com/explosion/spaCy.git
synced 2025-07-05 12:23:06 +03:00
Fix CharacterEmbed layer
This commit is contained in:
parent
97d3651574
commit
5ae8628571
|
@ -1,16 +1,18 @@
|
||||||
|
from typing import List
|
||||||
from thinc.api import Model
|
from thinc.api import Model
|
||||||
|
from thinc.types import Floats2d
|
||||||
|
from ..tokens import Doc
|
||||||
|
|
||||||
|
|
||||||
def CharacterEmbed(nM, nC):
|
def CharacterEmbed(nM: int, nC: int) -> Model[List[Doc], List[Floats2d]]:
|
||||||
# nM: Number of dimensions per character. nC: Number of characters.
|
# nM: Number of dimensions per character. nC: Number of characters.
|
||||||
nO = nM * nC if (nM is not None and nC is not None) else None
|
|
||||||
return Model(
|
return Model(
|
||||||
"charembed",
|
"charembed",
|
||||||
forward,
|
forward,
|
||||||
init=init,
|
init=init,
|
||||||
dims={"nM": nM, "nC": nC, "nO": nO, "nV": 256},
|
dims={"nM": nM, "nC": nC, "nO": nM * nC, "nV": 256},
|
||||||
params={"E": None},
|
params={"E": None},
|
||||||
).initialize()
|
)
|
||||||
|
|
||||||
|
|
||||||
def init(model, X=None, Y=None):
|
def init(model, X=None, Y=None):
|
||||||
|
|
Loading…
Reference in New Issue
Block a user