explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2024-09-21 19:39:13 +03:00
Code Issues Packages Projects Releases Wiki Activity

Expose the softmax layer in the tagger model, to allow setting tensors

Browse Source
This commit is contained in:
Matthew Honnibal 2017-11-03 11:19:51 +01:00
parent b30dd36179
commit b3264aa5f0
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
spacy/_ml.py
View File

@ -409,12 +409,14 @@ def build_tagger_model(nr_class, **cfg):
else: else:
tok2vec = Tok2Vec(token_vector_width, embed_size, tok2vec = Tok2Vec(token_vector_width, embed_size,
pretrained_dims=pretrained_dims) pretrained_dims=pretrained_dims)
softmax = with_flatten(Softmax(nr_class, token_vector_width))
model = ( model = (
tok2vec tok2vec
>> with_flatten(Softmax(nr_class, token_vector_width)) >> softmax
) )
model.nI = None model.nI = None
model.tok2vec = tok2vec model.tok2vec = tok2vec
model.softmax
return model return model