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

Change model structure to accept trf input

Browse Source
This commit is contained in:
richardpaulhudson 2023-01-25 11:36:33 +01:00
parent 872bda2f14
commit ba62f7eed9
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
spacy/ml/models/lemmatizer.py
View File

@ -37,12 +37,13 @@ def build_lemmatizer_model(
softmax = Softmax_v2( softmax = Softmax_v2(
nO, t2v_width, init_W=glorot_uniform_init, normalize_outputs=normalize nO, t2v_width, init_W=glorot_uniform_init, normalize_outputs=normalize
) )
model = tok2vec >> with_array(softmax)
if lowercasing: if lowercasing:
lowercasing_output = Sigmoid(1) lowercasing_output = Sigmoid(1)
sigmoid_appendage = ( sigmoid_appendage = (
Relu(lowercasing_relu_width) >> Dropout() >> lowercasing_output Relu(lowercasing_relu_width) >> Dropout() >> lowercasing_output
) )
model |= tok2vec >> with_array(sigmoid_appendage) model = tok2vec >> with_array(softmax | sigmoid_appendage)
model.set_ref("lowercasing_output", lowercasing_output) model.set_ref("lowercasing_output", lowercasing_output)
else:
model = tok2vec >> with_array(softmax)
return model return model