spaCy/spacy/pipeline
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
..
_edit_tree_internals
_parser_internals account for NER labels with a hyphen in the name (#10960) 2022-06-17 20:02:37 +01:00
legacy Refactor KB for easier customization (#11268) 2022-09-08 10:38:07 +02:00
__init__.py Add SpanRuler component (#9880) 2022-06-02 13:12:53 +02:00
attributeruler.py
dep_parser.pyx account for NER labels with a hyphen in the name (#10960) 2022-06-17 20:02:37 +01:00
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
entity_linker.py Refactor KB for easier customization (#11268) 2022-09-08 10:38:07 +02:00
entityruler.py Enable fuzzy text matching in Matcher (#11359) 2023-01-10 10:36:17 +01:00
functions.py
lemmatizer.py
morphologizer.pyx
multitask.pyx
ner.pyx account for NER labels with a hyphen in the name (#10960) 2022-06-17 20:02:37 +01:00
pipe.pxd
pipe.pyi
pipe.pyx Fix config validation failures caused by NVTX pipeline wrappers (#11460) 2022-09-12 14:55:41 +02:00
sentencizer.pyx
senter.pyx
span_ruler.py Enable fuzzy text matching in Matcher (#11359) 2023-01-10 10:36:17 +01:00
spancat.py Fix spancat for zero suggestions (#11860) 2022-12-02 09:33:52 +01:00
tagger.pyx
textcat.py Rename modified textcat scorer to v2 (#11971) 2022-12-29 14:01:08 +01:00
textcat_multilabel.py Improve score_cats for use with multiple textcat components (#11820) 2023-01-09 11:43:48 +01:00
tok2vec.py Prevent tok2vec to broadcast to listeners when predicting (#11385) 2022-09-12 15:36:48 +02:00
trainable_pipe.pxd
trainable_pipe.pyx Fix config validation failures caused by NVTX pipeline wrappers (#11460) 2022-09-12 14:55:41 +02:00
transition_parser.pxd Parser: use C saxpy/sgemm provided by the Ops implementation (#10773) 2022-05-27 11:20:52 +02:00
transition_parser.pyx precompute_hiddens/Parser: do not look up CPU ops (3.4) (#11069) 2022-07-05 10:53:42 +02:00