spaCy/spacy/tests/pipeline
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 Revert #4334 2019-09-29 17:32:12 +02:00
test_analysis.py Simplify pipe analysis 2020-08-01 13:40:06 +02:00
test_annotates_on_update.py Add training option to set annotations on update (#7767) 2021-04-26 16:53:53 +02:00
test_attributeruler.py Tidy up and auto-format 2021-01-05 13:41:53 +11:00
test_entity_linker.py KB & NEL to/from bytes (#8113) 2021-05-20 18:11:30 +10:00
test_entity_ruler.py Show warning if entity_ruler runs without patterns (#7807) 2021-06-04 17:37:38 +02:00
test_functions.py Add token_splitter component (#6726) 2021-01-17 19:54:41 +08:00
test_initialize.py Test with default value 2020-09-29 17:00:40 +02:00
test_lemmatizer.py Use warnings.warn instead of logger.warning 2021-06-04 17:44:08 +02:00
test_models.py Set up GPU CI testing (#7293) 2021-04-22 14:58:29 +02:00
test_morphologizer.py Handle unset token.morph in Morphologizer (#6704) 2021-01-15 17:20:10 +01:00
test_pipe_factories.py Resizable textcat (#7862) 2021-06-16 11:45:00 +02:00
test_pipe_methods.py Add training option to set annotations on update (#7767) 2021-04-26 16:53:53 +02:00
test_sentencizer.py Refactor Docs.is_ flags (#6044) 2020-09-17 00:14:01 +02:00
test_senter.py adding tests for trained models to ensure predict reproducibility 2020-10-13 21:07:13 +02:00
test_tagger.py Sync missing and misaligned values in Tagger loss (#6689) 2021-01-10 11:30:37 +11:00
test_textcat.py Resizable textcat (#7862) 2021-06-16 11:45:00 +02:00
test_tok2vec.py Ensemble textcat with listener (#8012) 2021-05-31 18:21:06 +10:00