def get_tok2vec_kwargs():
# This actually creates models, so seems best to put it in a function.
return {
"embed":
MultiHashEmbed(
width=32,
rows=[500, 500, 500],
attrs=["NORM", "PREFIX", "SHAPE"],
include_static_vectors=False,
),
"encode":
MaxoutWindowEncoder(width=32, depth=2, maxout_pieces=2, window_size=1),
}
def my_parser():
tok2vec = build_Tok2Vec_model(
MultiHashEmbed(
width=321,
attrs=["LOWER", "SHAPE"],
rows=[5432, 5432],
include_static_vectors=False,
),
MaxoutWindowEncoder(width=321, window_size=3, maxout_pieces=4, depth=2),
)
parser = build_tb_parser_model(
tok2vec=tok2vec,
state_type="parser",
extra_state_tokens=True,
hidden_width=65,
maxout_pieces=5,
use_upper=True,
)
return parser