File size: 1,297 Bytes
5b6f681 6609649 b3fbeb1 6609649 b3fbeb1 6609649 b3fbeb1 6609649 b3fbeb1 5b6f681 6609649 5b6f681 b3fbeb1 5b6f681 b3fbeb1 5b6f681 b3fbeb1 5b6f681 b3fbeb1 5b6f681 b3fbeb1 5b6f681 b3fbeb1 5b6f681 b3fbeb1 5b6f681 b3fbeb1 5b6f681 b3fbeb1 5b6f681 b3fbeb1 5b6f681 b3fbeb1 | 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 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 | ---
title: Transformer Sentiment Analysis
emoji: 🤖
colorFrom: blue
colorTo: purple
sdk: gradio
sdk_version: 4.44.0
app_file: app.py
pinned: false
license: apache-2.0
tags:
- sentiment-analysis
- transformers
- distilbert
- mlflow
- pytorch
models:
- MartinRodrigo/distilbert-sentiment-imdb
---
# 🤖 Transformer Sentiment Analysis
Advanced sentiment analysis using DistilBERT fine-tuned on IMDB dataset with MLflow experiment tracking.
## 🎯 Model Performance
- **Accuracy:** 80% on IMDB test set
- **F1 Score:** 0.7981
- **Model:** DistilBERT (66M parameters)
- **Speed:** ~100ms per prediction
## 🚀 Features
- Real-time sentiment analysis
- Batch text processing
- Confidence scores and probabilities
- Interactive visualizations
## 🔗 Links
- **Model Repository:** [MartinRodrigo/distilbert-sentiment-imdb](https://huggingface.co/MartinRodrigo/distilbert-sentiment-imdb)
- **GitHub:** [transformer-sentiment-analysis](https://github.com/mrdesautu/ransformer-sentiment-analysis)
## 💡 Usage
```python
from transformers import pipeline
classifier = pipeline("sentiment-analysis",
model="MartinRodrigo/distilbert-sentiment-imdb")
result = classifier("I love this movie!")
print(result)
```
Built with Transformers, MLflow, and Gradio 🚀 |