Categorium

Categorium is a Python package for building and training BERT-based text classification models, with preset parameters suited to small datasets. It includes six pre-trained models covering 27 categories each, and tools for testing language models.