spaCy/spacy/ml
Sofie Van Landeghem e796aab4b3
Resizable textcat (#7862)
* implement textcat resizing for TextCatCNN

* resizing textcat in-place

* simplify code

* ensure predictions for old textcat labels remain the same after resizing (WIP)

* fix for softmax

* store softmax as attr

* fix ensemble weight copy and cleanup

* restructure slightly

* adjust documentation, update tests and quickstart templates to use latest versions

* extend unit test slightly

* revert unnecessary edits

* fix typo

* ensemble architecture won't be resizable for now

* use resizable layer (WIP)

* revert using resizable layer

* resizable container while avoid shape inference trouble

* cleanup

* ensure model continues training after resizing

* use fill_b parameter

* use fill_defaults

* resize_layer callback

* format

* bump thinc to 8.0.4

* bump spacy-legacy to 3.0.6
2021-06-16 11:45:00 +02:00
..
models Resizable textcat (#7862) 2021-06-16 11:45:00 +02:00
__init__.py Tidy up and auto-format 2020-06-20 14:15:04 +02:00
_character_embed.py Register CharEmbed layer (#7805) 2021-04-19 18:39:34 +10:00
_precomputable_affine.py Tidy up, autoformat, add types 2020-07-25 15:01:15 +02:00
extract_ngrams.py register extract_ngrams layer (#8358) 2021-06-14 10:30:30 +02:00
featureextractor.py Fix import 2020-10-02 01:12:34 +02:00
parser_model.pxd The Parser is now a Pipe (2) (#5844) 2020-07-30 23:30:54 +02:00
parser_model.pyx The Parser is now a Pipe (2) (#5844) 2020-07-30 23:30:54 +02:00
staticvectors.py Set up GPU CI testing (#7293) 2021-04-22 14:58:29 +02:00
tb_framework.py Resizable textcat (#7862) 2021-06-16 11:45:00 +02:00