explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-01-27 09:44:36 +03:00
Code Issues Packages Projects Releases Wiki Activity

Wire up tb_framework to new parser model

Browse Source
This commit is contained in:
Matthew Honnibal 2021-10-25 12:50:20 +02:00
parent 0279aa036a
commit 71abe2e42d
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
spacy/ml/tb_framework.py
View File

@ -1,5 +1,5 @@
from typing import List, Tuple, Any, Optional
from thinc.api import Ops, Model, normal_init
from thinc.api import Ops, Model, normal_init, chain, list2array, Linear
from thinc.types import Floats1d, Floats2d, Floats3d, Ints2d, Floats4d
from ..tokens.doc import Doc
@ -20,11 +20,15 @@ def TransitionModel(
"""Set up a transition-based parsing model, using a maxout hidden
layer and a linear output layer.
"""
t2v_width = tok2vec.get_dim("nO") if tok2vec.has_dim("nO") else None
tok2vec_projected = chain(tok2vec, list2array(), Linear(hidden_width, t2v_width))
tok2vec_projected.set_dim("nO", hidden_width)
return Model(
name="parser_model",
forward=forward,
init=init,
layers=[tok2vec],
layers=[tok2vec_projected],
refs={"tok2vec": tok2vec},
params={
"lower_W": None, # Floats2d W for the hidden layer