README.md

metadata

pipeline_tag: text-classification
metrics:
  - accuracy

BERT Text Classification Model

This is a simple demo model for text classification using BERT.

Usage

To use the model, you can call the classify_text function with a text input, and it will return the predicted class label.

text = "This is a positive review."
predicted_class = classify_text(text)
print("Predicted class:", predicted_class)

Devlin, J., Chang, M. W., Lee, K., & Toutanova, K. (2018). BERT: Pre-training of Deep Bidirectional Transformers for Language Understanding. arXiv preprint arXiv:1810.04805.