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	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	Tidy up and auto-format	2021-07-18 15:44:56 +10:00
test_attributeruler.py	Refactor scoring methods to use registered functions (#8766)	2021-08-10 15:13:39 +02:00
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	Add doc_cleaner component (#9659)	2021-11-23 15:33:33 +01:00
test_initialize.py	Test with default value	2020-09-29 17:00:40 +02:00
test_lemmatizer.py	Tidy up and auto-format	2021-07-18 15:44:56 +10:00
test_models.py	Tidy up code	2021-06-28 12:08:15 +02:00
test_morphologizer.py	removing print statements from the test suite (#10712)	2022-04-27 09:14:25 +02:00
test_pipe_factories.py	Auto-format code with black (#10795)	2022-05-13 19:02:08 +02:00
test_pipe_methods.py	Revert disable/disabled merging behavior (#11745)	2022-11-08 14:58:10 +01:00
test_sentencizer.py	Refactor Docs.is_ flags (#6044)	2020-09-17 00:14:01 +02:00
test_senter.py	Add Pipe.hide_labels to omit labels from pipeline meta (#10175)	2022-02-05 17:59:24 +01:00
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	Migrate regression tests into the main test suite (#9655)	2021-12-04 20:34:48 +01:00
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