File size: 773 Bytes
b998ba1 | 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 | # BERT Fine-Tuned Classifier
This model was fine-tuned using BERT for text classification.
## Training Details
- **Dataset:** [Your Dataset Name] # Replace with actual dataset
- **Hyperparameters:**
- Epochs: 3
- Batch Size: 8
- Learning Rate: 5e-5
- Weight Decay: 0.01
## Evaluation Metrics
- **Loss:** 0.5719
- **Accuracy:** 87.74%
- **Precision:** 86.47%
- **Recall:** 89.56%
- **F1-score:** 87.99%
## Model Usage
To use this model with `transformers`:
```python
from transformers import pipeline
model_name = "Tryfonas/fine-tuned-bert-classifier-bds24" # Update with your repo name
classifier = pipeline("text-classification", model=model_name)
result = classifier("This is an example text")
print(result) |