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:
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)