cornell-movie-review-data/rotten_tomatoes
Viewer • Updated • 10.7k • 73.6k • 112
How to use flowfree/bert-finetuned-rottentomatoes with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="flowfree/bert-finetuned-rottentomatoes") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("flowfree/bert-finetuned-rottentomatoes")
model = AutoModelForSequenceClassification.from_pretrained("flowfree/bert-finetuned-rottentomatoes")This model is a fine-tuned version of bert-base-cased on the rotten_tomatoes dataset. It achieves the following results on the evaluation set:
More information needed
More information needed
More information needed
The following hyperparameters were used during training:
| Training Loss | Epoch | Step | Validation Loss | Accuracy |
|---|---|---|---|---|
| 0.1626 | 1.0 | 1067 | 0.8012 | 0.8340 |
| 0.1048 | 2.0 | 2134 | 0.9137 | 0.8405 |
| 0.0472 | 3.0 | 3201 | 0.9971 | 0.8443 |