mirror of https://github.com/explosion/spaCy.git
Adjust textcat model
This commit is contained in:
parent
9e1b11dd81
commit
c99a653070
|
@ -158,7 +158,6 @@ def build_text_classifier_lowdata(width, pretrained_vectors, dropout, nO=None):
|
|||
model = (
|
||||
StaticVectors(width)
|
||||
>> list2ragged()
|
||||
>> with_ragged(0, Linear(width, vector_dim))
|
||||
>> ParametricAttention(width)
|
||||
>> reduce_sum()
|
||||
>> residual(Relu(width, width)) ** 2
|
||||
|
|
Loading…
Reference in New Issue