explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-08-24 22:14:56 +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(
nO, t2v_width, init_W=glorot_uniform_init, normalize_outputs=normalize
)
model = tok2vec >> with_array(softmax)
if lowercasing:
lowercasing_output = Sigmoid(1)
sigmoid_appendage = (
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)
else:
model = tok2vec >> with_array(softmax)
return model