Add nO attribute to TextCategorizer model

This commit is contained in:
Matthew Honnibal 2017-10-04 16:07:30 +02:00
parent f8a0614527
commit bd8e84998a

View File

@ -642,7 +642,7 @@ def build_text_classifier(nr_class, width=64, **cfg):
>> zero_init(Affine(nr_class, nr_class*2, drop_factor=0.0))
>> logistic
)
model.nO = nr_class
model.lsuv = False
return model