DistilBERT Fine-tuned for AG News Classification
This model is a fine-tuned version of distilbert-ag-news-classifier on the AG News dataset.
Model Details
- Model Type: DistilBERT for Sequence Classification
- Dataset: AG News (sampled)
- Training Technique: Fine-tuning for 1 epoch
Performance Metrics
- Accuracy: 0.8870
- Precision: 0.8864
- Recall: 0.8870
- F1 Score: 0.8861
Classes
- 0: World
- 1: Sports
- 2: Business
- 3: Sci/Tech
- Downloads last month
- 1