spaCy/spacy/ml/models
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
..
__init__.py pretrain architectures (#6451) 2020-12-08 14:41:03 +08:00
entity_linker.py make EntityLinker robust for nO=None (#7930) 2021-05-06 18:14:47 +10:00
multi_task.py applying suggestion to avoid mypy errors (#8265) 2021-06-02 19:25:30 +10:00
parser.py consistently use registry as callable 2021-03-02 17:56:28 +01:00
tagger.py consistently use registry as callable 2021-03-02 17:56:28 +01:00
textcat.py Resizable textcat (#7862) 2021-06-16 11:45:00 +02:00
tok2vec.py Replace negative rows with 0 in StaticVectors (#7674) 2021-04-22 18:04:15 +10:00