Can I add custom classifier to NER pipeline?

I need to resolve ner chunks to codes, in my case map identified health problem in the medical text to a diagnosis code. I can train a custom classifier to map ner chunk, but can I add this classifier as a stage / later to the NLP NER pipeline? Thanks!

You will need to do the steps of the pipeline manually, probably: so call the tokenizer on your text, the pretained model on the outputs and then apply your custom classifier.

1 Like