Downloads · 30 days
12
2% of all-time downloads
bert-base-uncased-emotion
sabersol/bert-base-uncased-emotion
bert-base-uncased-emotion is a text classification model from sabersol. Use it when you need a label for a piece of text. It is set up for transformers. The card lists the license as cc-by-nc-sa-4.0.
Fine-tuned bert-base-uncased on the emotions dataset
All-time downloads
762
Public
Repo size
1.8 GB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin438 MB · 100%
At a glance
- Task
- Text Classification
- Library
- transformers
- License
- cc-by-nc-sa-4.0
- Model type
- bert
- Access
- Public
- Created
- May 19, 2022
- Updated
- May 27, 2022
- SHA
- f80494fa
Try a prompt
Model card
From the Hugging Face model README
CITDA:
Fine-tuned bert-base-uncased on the emotions dataset
Demo Notebook: https://colab.research.google.com/drive/10ZCFvlf2UV3FjU4ymf4OoipQvqHbIItG?usp=sharing
Packages
- Install
torch - Also,
pip install transformers datasets scikit-learn wandb seaborn python-dotenv
Train
- Rename
.env.exampleto.envand set an API key from wandb - You can adjust model parameters in the
explainableai.pyfile. - The model (
pytorch_model.bin) is a based on thebert-base-uncasedand already trained on theemotionsdataset. To re-produce the training runfinetune-emotions.py. You can change the base model, or the dataset by changing that file's code.
Example
Run example.py
Train
The model is already trained on bert-base-uncased with the emotions dataset. However, you can change parameters and re-fine-tune the model by running finetune-emotions.py.
- Task
- Text Classification
- Library
- transformers
- Type
- bert
- License
- cc-by-nc-sa-4.0
- Created
- May 19, 2022
- Updated
- May 27, 2022