Use last hidden as feature
@@ -33,7 +33,7 @@ class PyTT_TextCategorizer(spacy.pipeline.TextCategorizer):
RETURNS (thinc.neural.Model): The model.
"""
return chain(
- get_pytt_class_tokens,
+ get_pytt_last_hidden,
flatten_add_lengths,
Pooling(mean_pool),
Softmax(nr_class, cfg["token_vector_width"]),
