Downloads · 30 days
54
1% of all-time downloads
bert-base-uncased-ag-news
nateraw/bert-base-uncased-ag-news
bert-base-uncased-ag-news is a text classification model from nateraw. Use it when you need a label for a piece of text. It is set up for transformers. The card lists the license as mit.
bert-base-uncased finetuned on the AG News dataset using PyTorch Lightning. Sequence length 128, learning rate 2e-5, batch size 32, 4 T4 GPUs, 4 epochs. The code can be found here
All-time downloads
9.6K
Public
Repo size
1.3 GB
Likes
3
Public
Repo makeup
Click a slice to open those files.
.bin438 MB · 50%
At a glance
- Task
- Text Classification
- Library
- transformers
- License
- mit
- Model type
- bert
- Access
- Public
- Created
- Mar 2, 2022
- Updated
- Sep 22, 2021
- SHA
- 5d0540f6
Try a prompt
Datasets
Model card
From the Hugging Face model README
bert-base-uncased-ag-news
Model description
bert-base-uncased finetuned on the AG News dataset using PyTorch Lightning. Sequence length 128, learning rate 2e-5, batch size 32, 4 T4 GPUs, 4 epochs. The code can be found here
Limitations and bias
- Not the best model...
Training data
Data came from HuggingFace's datasets package. The data can be viewed on nlp viewer.
Training procedure
...
Eval results
...
- Task
- Text Classification
- Library
- transformers
- Type
- bert
- License
- mit
- Languages
- en
- Created
- Mar 2, 2022
- Updated
- Sep 22, 2021