File size: 632 Bytes
00491de | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 | ---
tags:
- text-classification
- sentiment-analysis
- distilbert
license: apache-2.0
---
# 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
```python
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
|