| 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 | |