Mention Classifier
Category prediction model This repo contains AllenNLP model for prediction of Named Entity categories by its mentions.
natural-language-processing named-entity-recognition mention-classification
Objectives & Highlights

Mention classification lays between Named Entity Recognition and Entity Linking. - It provides much more detailed information about an object than NER does. - At the same time, it does not require storing and maintaining any knowledge base with known objects. - The Classifier able to work with absolutely new objects never appeared in the train set. - The neural net is able to look at up to five mentions simultaneously. It combines evidence more efficient than it could be achieved by averaging individual predictions.

Takeaways & Next Steps

- Support for other languages - Support for other entities

Don't forget to add the tag @generall in your comments.

Lead ML Engineer
Share this project
Similar projects
Custom Classifier on Top of Bert-like Language Model
Take pre-trained language model and build custom classifier on top of it.
Finetune: Scikit-learn Style Model Finetuning for NLP
Finetune is a library that allows users to leverage state-of-the-art pretrained NLP models for a wide variety of downstream tasks.
Transformer OCR
Rectification-free OCR using spatial attention from Transformers.
Using Different Decoding Methods for LM with Transformers
A look at different decoding methods for generate subsequent tokens in language modeling.