Downloads · 30 days
1
0% of all-time downloads
axilion/RoBERTa-movie-sentiment-analyzer
RoBERTa-movie-sentiment-analyzer is a machine learning model from axilion. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
Fine-tuned from roberta-base on the IMDb movie-review dataset • Accuracy: 94% on validation • Hyper-parameters: 1 epochs · lr 1e-05 · batch 8 · maxlen 256
Downloads · 30 days
1
0% of all-time downloads
All-time downloads
220
Public
Parameters
125M
499 MB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors499 MB · 99%
From the Hugging Face model README
Fine-tuned from roberta-base on the IMDb movie-review dataset
• Accuracy: 94% on validation
• Hyper-parameters: 1 epochs · lr 1e-05 · batch 8 · max_len 256
from transformers import pipeline
clf = pipeline("sentiment-analysis",
model="your-username/roberta-imdb-sentiment",
tokenizer="your-username/roberta-imdb-sentiment")
print(clf("Such a wonderful movie!"))