Skip to main content

Text classifier, based on the BERT and a Bayesian neural network, which can train on small labeled texts and doubt its decision

Project description

Text classifier, based on the BERT and a Bayesian neural network, which can train on small labeled texts and doubt its decision.

The goal of this project is developing of simple and power text classifier based on transfer learning and bayesian neural networks.

A transfer learning (particulary, well-known BERT model) helps to generate special contextual embeddings for text tokens, which provide a better discrimination ability in feature space, than classical word embeddings. Therefore we can use smaller labeled data for training of final classifier.

Bayesian neural network in final classifier models uncertainty in data, owing to this fact probabilities of recognized classes returned by this network are more fair, and bayesian neural network is more robust to overfitting.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Files for impartial-text-cls, version 0.0.4
Filename, size File type Python version Upload date Hashes
Filename, size impartial-text-cls-0.0.4.tar.gz (26.0 kB) File type Source Python version None Upload date Hashes View hashes

Supported by

Elastic Elastic Search Pingdom Pingdom Monitoring Google Google BigQuery Sentry Sentry Error logging AWS AWS Cloud computing DataDog DataDog Monitoring Fastly Fastly CDN DigiCert DigiCert EV certificate StatusPage StatusPage Status page