Instructions to use Wolverine001/bert_finetuned_senti with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use Wolverine001/bert_finetuned_senti with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="Wolverine001/bert_finetuned_senti")# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("Wolverine001/bert_finetuned_senti") model = AutoModelForSequenceClassification.from_pretrained("Wolverine001/bert_finetuned_senti") - Notebooks
- Google Colab
- Kaggle
Fine-Tuned BERT Sentiment Model
This model was fine-tuned for sentiment classification.
- Pre-trained model used: google-bert/bert-base-uncased.
- Dataset used: twitter-sentiment.
- max_length = 128
- batch_size = 8
- learning_rate = 1e-4
- epochs = 3
Evaluation Results
π Before Fine-Tuning
Accuracy: 0.4046
| Class | Precision | Recall | F1-Score | Support |
|---|---|---|---|---|
| Negative | 0.00 | 0.00 | 0.00 | 1001 |
| Neutral | 0.40 | 1.00 | 0.58 | 1430 |
| Positive | 0.00 | 0.00 | 0.00 | 1103 |
| Macro Avg | 0.13 | 0.33 | 0.19 | 3534 |
| Weighted Avg | 0.16 | 0.40 | 0.23 | 3534 |
β After Fine-Tuning
Accuracy: 0.6095
| Class | Precision | Recall | F1-Score | Support |
|---|---|---|---|---|
| Negative | 0.82 | 0.29 | 0.42 | 1001 |
| Neutral | 0.51 | 0.89 | 0.65 | 1430 |
| Positive | 0.85 | 0.54 | 0.66 | 1103 |
| Macro Avg | 0.73 | 0.57 | 0.58 | 3534 |
| Weighted Avg | 0.70 | 0.61 | 0.59 | 3534 |
You can download the model from Hugging Face.
- Downloads last month
- 2
Model tree for Wolverine001/bert_finetuned_senti
Base model
google-bert/bert-base-uncased