Tryfonas's picture
Uploading fine-tuned model
b998ba1 verified
# 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)