gender-classifier
Collection
7 items • Updated
How to use IsGarrido/gender_classifier_en_modernbert_base with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="IsGarrido/gender_classifier_en_modernbert_base") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("IsGarrido/gender_classifier_en_modernbert_base")
model = AutoModelForSequenceClassification.from_pretrained("IsGarrido/gender_classifier_en_modernbert_base")This model was fine-tuned to classify text into: male, female, neutral
| Metric | Value |
|---|---|
| loss | 0.2923 |
| accuracy | 0.9330 |
| f1 | 0.9326 |
| precision | 0.9332 |
| recall | 0.9332 |
| runtime | 27.4919 |
| samples_per_second | 218.2460 |
| steps_per_second | 3.4190 |
| 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
answerdotai/ModernBERT-base