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

Pass settings better from parser

Browse Source
This commit is contained in:
Matthw Honnibal 2019-10-22 03:26:43 +02:00
parent ab7f85dfa2
commit 1dce86c555
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
spacy/syntax/nn_parser.pyx
View File

@ -57,6 +57,7 @@ cdef class Parser:
cfg.get('subword_features', True)) cfg.get('subword_features', True))
conv_depth = util.env_opt('conv_depth', cfg.get('conv_depth', 4)) conv_depth = util.env_opt('conv_depth', cfg.get('conv_depth', 4))
conv_window = util.env_opt('conv_window', cfg.get('conv_depth', 1)) conv_window = util.env_opt('conv_window', cfg.get('conv_depth', 1))
t2v_pieces = util.env_opt('cnn_maxout_pieces', cfg.get('cnn_maxout_pieces', 3))
bilstm_depth = util.env_opt('bilstm_depth', cfg.get('bilstm_depth', 0)) bilstm_depth = util.env_opt('bilstm_depth', cfg.get('bilstm_depth', 0))
self_attn_depth = util.env_opt('self_attn_depth', cfg.get('self_attn_depth', 0)) self_attn_depth = util.env_opt('self_attn_depth', cfg.get('self_attn_depth', 0))
if depth not in (0, 1): if depth not in (0, 1):
@ -73,6 +74,7 @@ cdef class Parser:
tok2vec = Tok2Vec(token_vector_width, embed_size, tok2vec = Tok2Vec(token_vector_width, embed_size,
conv_depth=conv_depth, conv_depth=conv_depth,
conv_window=conv_window, conv_window=conv_window,
cnn_maxout_pieces=t2v_pieces,
subword_features=subword_features, subword_features=subword_features,
pretrained_vectors=pretrained_vectors, pretrained_vectors=pretrained_vectors,
bilstm_depth=bilstm_depth, bilstm_depth=bilstm_depth,
@ -101,7 +103,8 @@ cdef class Parser:
'self_attn_depth': self_attn_depth, 'self_attn_depth': self_attn_depth,
'conv_depth': conv_depth, 'conv_depth': conv_depth,
'conv_window': conv_window, 'conv_window': conv_window,
'embed_size': embed_size 'embed_size': embed_size,
'cnn_maxout_pieces': t2v_pieces
} }
return ParserModel(tok2vec, lower, upper), cfg return ParserModel(tok2vec, lower, upper), cfg