Downloads · 30 days
403
0% of all-time downloads
roberta-base-finetuned-yelp-polarity
VictorSanh/roberta-base-finetuned-yelp-polarity
roberta-base-finetuned-yelp-polarity is a text classification model from VictorSanh. Use it when you need a label for a piece of text. It is set up for transformers.
This is a RoBERTa-base checkpoint fine-tuned on binary sentiment classifcation from Yelp polarity. It gets 98.08% accuracy on the test set.
All-time downloads
642K
Public
Parameters
125M
1.5 GB on disk
Likes
5
Public
Repo makeup
Click a slice to open those files.
.bin501 MB · 33%
At a glance
- Task
- Text Classification
- Library
- transformers
- Model type
- roberta
- Access
- Public
- Created
- Mar 2, 2022
- Updated
- Mar 17, 2023
- SHA
- 709b46dd
Try a prompt
Datasets
Model card
From the Hugging Face model README
RoBERTa-base-finetuned-yelp-polarity
This is a RoBERTa-base checkpoint fine-tuned on binary sentiment classifcation from Yelp polarity. It gets 98.08% accuracy on the test set.
Hyper-parameters
We used the following hyper-parameters to train the model on one GPU:
num_train_epochs = 2.0
learning_rate = 1e-05
weight_decay = 0.0
adam_epsilon = 1e-08
max_grad_norm = 1.0
per_device_train_batch_size = 32
gradient_accumulation_steps = 1
warmup_steps = 3500
seed = 42
- Task
- Text Classification
- Library
- transformers
- Type
- roberta
- Languages
- en
- Created
- Mar 2, 2022
- Updated
- Mar 17, 2023