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

Fix definition of morphology model

This commit is contained in:
Matthew Honnibal 2018-09-25 22:12:32 +02:00
parent fb0abddd9e
commit 53eb96db09
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
spacy/_ml.py
View File

@ -524,8 +524,8 @@ def build_morphologizer_model(class_nums, **cfg):
tok2vec = Tok2Vec(token_vector_width, embed_size,
subword_features=subword_features,
pretrained_vectors=pretrained_vectors)
softmax = with_flatten(
MultiSoftmax(class_nums, token_vector_width))
softmax = with_flatten(MultiSoftmax(class_nums, token_vector_width))
softmax.out_sizes = class_nums
model = (
tok2vec
>> softmax