IsGarrido/gender_classifier_es_roberta_large

Gender Classifier (Fine-tuned illuin/roberta-large-bne)

This model was fine-tuned to classify text into: male, female, neutral

Performance Metrics

Metric	Value
loss	0.4650
accuracy	0.8857
f1	0.8856
precision	0.8856
recall	0.8857
runtime	10.4045
samples_per_second	576.6720
steps_per_second	36.0420
epoch	3.0000

Hyperparameters

• Batch Size: 16
• Learning Rate: 5e-05
• Epochs: 3
• Weight Decay: 0.01
• Mixed Precision (FP16): True

Quick Usage

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.'))