Rooba
/
spaCy
mirror of https://github.com/explosion/spaCy.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Format

This commit is contained in:
Matthew Honnibal 2020-10-05 15:29:49 +02:00
parent 7d93575f35
commit eb9ba61517
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
spacy/ml/models/tok2vec.py
View File

@ -103,17 +103,18 @@ def MultiHashEmbed_v1(
here as a temporary compatibility.""" here as a temporary compatibility."""
return MultiHashEmbed( return MultiHashEmbed(
width=width, width=width,
rows=rows, rows=rows,
attrs=[NORM, PREFIX, SUFFIX, SHAPE] if also_embed_subwords else [NORM], attrs=[NORM, PREFIX, SUFFIX, SHAPE] if also_embed_subwords else [NORM],
include_static_vectors=also_use_static_vectors include_static_vectors=also_use_static_vectors,
) )
@registry.architectures.register("spacy.MultiHashEmbed.v2") @registry.architectures.register("spacy.MultiHashEmbed.v2")
def MultiHashEmbed( def MultiHashEmbed(
width: int, width: int,
rows: int, rows: int,
attrs: Union[List[Union[str, int]], Dict[Union[str, int], float]], attrs: Union[List[Union[str, int]], Dict[Union[str, int], float]],
include_static_vectors: bool include_static_vectors: bool,
) -> Model[List[Doc], List[Floats2d]]: ) -> Model[List[Doc], List[Floats2d]]:
"""Construct an embedding layer that separately embeds a number of lexical """Construct an embedding layer that separately embeds a number of lexical
attributes using hash embedding, concatenates the results, and passes it attributes using hash embedding, concatenates the results, and passes it