stanfordnlp/imdb
Viewer • Updated • 100k • 178k • 370
How to use versae/gzipbert_imdb with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="versae/gzipbert_imdb", trust_remote_code=True) # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("versae/gzipbert_imdb", trust_remote_code=True)
model = AutoModelForSequenceClassification.from_pretrained("versae/gzipbert_imdb", trust_remote_code=True)This model is a fine-tuned version of versae/gzip-bert on the imdb 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.6987 | 1.0 | 1563 | 0.6937 | 0.5 |
| 0.7007 | 2.0 | 3126 | 0.6932 | 0.5 |
| 0.6993 | 3.0 | 4689 | 0.6932 | 0.5 |
| 0.6971 | 4.0 | 6252 | 0.6934 | 0.5 |
| 0.6966 | 5.0 | 7815 | 0.6932 | 0.5 |