| language: en | |
| license: apache-2.0 | |
| tags: | |
| - sentiment-analysis | |
| - text-classification | |
| - distilbert | |
| datasets: | |
| - imdb | |
| metrics: | |
| - accuracy | |
| - f1 | |
| pipeline_tag: text-classification | |
| # sentiment-tutorial | |
| Fine-tuned distilbert-base-uncased for binary sentiment classification. | |
| ## Intended Use | |
| Classify English text as positive or negative. | |
| ## Training Procedure | |
| - Base model: distilbert-base-uncased | |
| - Epochs: 2 | |
| - Learning rate: 2e-5 | |
| - Batch size: 32 | |
| - Max length: 128 | |
| ## Evaluation Results | |
| Accuracy: 0.869 | |
| Precision: 0.878 | |
| Recall: 0.858 | |
| F1: 0.868 | |
| ## Limitations | |
| - Binary classification only | |
| - English only | |
| - Movie reviews domain | |
| ## Usage | |
| from transformers import pipeline | |
| classifier = pipeline( | |
| "sentiment-analysis", | |
| model="Rameen191/sentiment-tutorial" | |
| ) | |
| classifier("This was a great experience!") | |