| license: mit | |
| language: | |
| - en | |
| base_model: | |
| - google-bert/bert-base-uncased | |
| pipeline_tag: text-classification | |
| # bert-sentiment-imdb | |
| This is a fine-tuned BERT model for sentiment analysis on the IMDB dataset. | |
| - **Base Model**: `bert-base-uncased` | |
| - **Dataset**: IMDB | |
| - **Task**: Sentiment Analysis (Binary) | |
| - **Usage**: | |
| ```python | |
| from transformers import pipeline | |
| classifier = pipeline("sentiment-analysis"model="hmuegyi/review-classifier") | |
| # Test | |
| result = classifier("This movie is nice!") | |
| print(result) | |
| ``` |