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

Expose hyper-param as keyword arg

Browse Source
This commit is contained in:
Matthew Honnibal 2017-07-22 20:14:37 +02:00
parent f5de8deeec
commit ded0df5e2f
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
spacy/_ml.py
View File

@ -411,11 +411,12 @@ def uniqued(layer, column=0):
def build_text_classifier(nr_class, width=64, **cfg):
nr_vector = cfg.get('nr_vector', 1000)
with Model.define_operators({'>>': chain, '+': add, '|': concatenate, '**': clone}):
embed_lower = HashEmbed(width, 1000, column=1)
embed_prefix = HashEmbed(width//2, 1000, column=2)
embed_suffix = HashEmbed(width//2, 1000, column=3)
embed_shape = HashEmbed(width//2, 1000, column=4)
embed_lower = HashEmbed(width, nr_vector, column=1)
embed_prefix = HashEmbed(width//2, nr_vector, column=2)
embed_suffix = HashEmbed(width//2, nr_vector, column=3)
embed_shape = HashEmbed(width//2, nr_vector, column=4)
model = (
FeatureExtracter([ORTH, LOWER, PREFIX, SUFFIX, SHAPE])