My Awesome Model
This is a fine-tuned DistilBERT model for sentiment analysis on the IMDb dataset. It classifies movie reviews as positive or negative.
Usage
from transformers import AutoModelForSequenceClassification, AutoTokenizer
model = AutoModelForSequenceClassification.from_pretrained("your_username/testing_the_water")
tokenizer = AutoTokenizer.from_pretrained("your_username/testing_the_water")
Training
- Dataset: IMDb (subset of 1,000 samples)
- Epochs: 1
- Batch Size: 8
- Downloads last month
- 8