---

language: en license: apache-2.0 tags:

• text-classification
• bert
• pytorch datasets:
• custom metrics:
• accuracy
• f1

---

Text Classification Model

This model classifies text into 5 categories:

• Business
• Technology
• Sports
• Politics
• Entertainment

Usage

from transformers import BertTokenizer, BertForSequenceClassification

tokenizer = BertTokenizer.from_pretrained('myekini/text-classification-bert')
model = BertForSequenceClassification.from_pretrained('myekini/text-classification-bert')

Model Details

• Model Type: BERT for Sequence Classification
• Language: English
• License: Apache 2.0
• Training Data: Custom dataset