cornell-movie-review-data/rotten_tomatoes
Viewer • Updated • 10.7k • 73.4k • 112
How to use aj-data/rotten_tomatoes_dataset with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="aj-data/rotten_tomatoes_dataset") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("aj-data/rotten_tomatoes_dataset")
model = AutoModelForSequenceClassification.from_pretrained("aj-data/rotten_tomatoes_dataset")This model is a fine-tuned version of bert-base-uncased 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.3923 | 1.0 | 1067 | 0.3781 | 0.8480 |
| 0.2186 | 2.0 | 2134 | 0.5862 | 0.8518 |
| 0.0747 | 3.0 | 3201 | 0.7857 | 0.8630 |