categorium

A package for creating and training language models for text classification based on BERT. The package includes pre-trained models and a feature for testing the trained models.


Keywords
python, classification, text, Categorization, textCategorization, textclassification, BERT
Install
pip install categorium==0.2.1