🚀 Model Card for Model ID
This is a DistilBERT-based model fine-tuned for text classification on the AG News dataset, categorizing news articles into four categories.
🚀 Quick Start
Use the code below to get started with the model.
[More Information Needed]
✨ Features
- This is a DistilBERT-based model fine-tuned for text classification on the AG News dataset.
- The model categorizes news articles into four categories: World, Sports, Business, and Science/Technology.
- DistilBERT is a smaller and faster variant of BERT, making it more efficient for real-time applications while maintaining strong performance.
📚 Documentation
Model Details
- Developed by: Ranudee Fernando
- Model type: DistilBERT-based text classifier
- Language(s) (NLP): English (EN)
- License: Apache 2.0
- Finetuned from model [optional]: Fine-tuned from
distilbert-base-uncased
Property |
Details |
Model Type |
DistilBERT-based text classifier |
Language |
English (EN) |
License |
Apache 2.0 |
Finetuned from model |
Fine-tuned from distilbert-base-uncased |
Model Sources [optional]
- Repository: [More Information Needed]
- Paper [optional]: [More Information Needed]
- Demo [optional]: [More Information Needed]
Uses
Direct Use
[More Information Needed]
Downstream Use [optional]
[More Information Needed]
Out-of-Scope Use
[More Information Needed]
Bias, Risks, and Limitations
[More Information Needed]
Recommendations
Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
Training Details
Training Data
[More Information Needed]
Training Procedure
Preprocessing [optional]
[More Information Needed]
Training Hyperparameters
- Training regime: [More Information Needed]
Speeds, Sizes, Times [optional]
[More Information Needed]
Evaluation
Testing Data, Factors & Metrics
Testing Data
[More Information Needed]
Factors
[More Information Needed]
Metrics
[More Information Needed]
Results
[More Information Needed]
Summary
[More Information Needed]
Model Examination [optional]
[More Information Needed]
Environmental Impact
Carbon emissions can be estimated using the Machine Learning Impact calculator presented in Lacoste et al. (2019).
- Hardware Type: [More Information Needed]
- Hours used: [More Information Needed]
- Cloud Provider: [More Information Needed]
- Compute Region: [More Information Needed]
- Carbon Emitted: [More Information Needed]
Technical Specifications [optional]
Model Architecture and Objective
[More Information Needed]
Compute Infrastructure
Hardware
[More Information Needed]
Software
[More Information Needed]
Citation [optional]
BibTeX:
[More Information Needed]
APA:
[More Information Needed]
Glossary [optional]
[More Information Needed]
More Information [optional]
[More Information Needed]
Model Card Authors [optional]
[More Information Needed]
Model Card Contact
[More Information Needed]