gender-classifier
Collection
7 items
•
Updated
This model was fine-tuned to classify text into: male, female, neutral
| Metric | Value |
|---|---|
| loss | 0.3096 |
| accuracy | 0.9313 |
| f1 | 0.9309 |
| precision | 0.9318 |
| recall | 0.9315 |
| runtime | 2.9479 |
| samples_per_second | 2035.3350 |
| steps_per_second | 31.8870 |
| epoch | 3.0000 |
from transformers import pipeline
# Load the model directly from this folder or HF Hub
classifier = pipeline('text-classification', model='.')
print(classifier('She is a great engineer.'))
Base model
andersonbcdefg/distilmodernbert