metadata
language: en
license: apache-2.0
tags:
- text-classification
- ag-news
- distilbert
datasets:
- ag_news
metrics:
- accuracy
- precision
- recall
- f1
base_model: distilbert-ag-news-classifier
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