Richard Hudson f9e020dd67 Fix speed problem with top_k>1 on CPU in edit tree lemmatizer (#12017) ... * Refactor _scores2guesses * Handle arrays on GPU * Convert argmax result to raw integer Co-authored-by: Madeesh Kannan <shadeMe@users.noreply.github.com> * Use NumpyOps() to copy data to CPU Co-authored-by: Madeesh Kannan <shadeMe@users.noreply.github.com> * Changes based on review comments * Use different _scores2guesses depending on tree_k * Add tests for corner cases * Add empty line for consistency * Improve naming Co-authored-by: Daniël de Kok <me@github.danieldk.eu> * Improve naming Co-authored-by: Daniël de Kok <me@github.danieldk.eu> Co-authored-by: Madeesh Kannan <shadeMe@users.noreply.github.com> Co-authored-by: Daniël de Kok <me@github.danieldk.eu>		2023-01-20 19:34:11 +01:00
..
__init__.py
test_analysis.py
test_annotates_on_update.py
test_attributeruler.py
test_edit_tree_lemmatizer.py	Fix speed problem with top_k>1 on CPU in edit tree lemmatizer (#12017)	2023-01-20 19:34:11 +01:00
test_entity_linker.py	Handle Docs with no entities in EntityLinker (#11640)	2022-10-28 10:25:34 +02:00
test_entity_ruler.py	Enable fuzzy text matching in Matcher (#11359)	2023-01-10 10:36:17 +01:00
test_functions.py
test_initialize.py
test_lemmatizer.py
test_models.py
test_morphologizer.py
test_pipe_factories.py
test_pipe_methods.py	Revert disable/disabled merging behavior (#11745)	2022-11-08 14:58:10 +01:00
test_sentencizer.py
test_senter.py
test_span_ruler.py	Add SpanRuler component (#9880)	2022-06-02 13:12:53 +02:00
test_spancat.py	Fix spancat for zero suggestions (#11860)	2022-12-02 09:33:52 +01:00
test_tagger.py
test_textcat.py	Improve score_cats for use with multiple textcat components (#11820)	2023-01-09 11:43:48 +01:00
test_tok2vec.py	Auto-format code with black (#11687)	2022-10-21 11:54:17 +02:00