YAML Metadata Warning: empty or missing yaml metadata in repo card (https://huggingface.co/docs/hub/model-cards#model-card-metadata)

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)
Downloads last month
1
Safetensors
Model size
0.1B params
Tensor type
F32
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Space using Tryfonas/fine-tuned-bert-classifier-bds24 1