IsGarrido/gender_classifier_multi_xlmroberta_large

Gender Classifier (Fine-tuned xlm-roberta-large)

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

Performance Metrics

Metric	Value
loss	0.4028
accuracy	0.8830
f1	0.8827
precision	0.8853
recall	0.8831
runtime	43.8849
samples_per_second	273.4430
steps_per_second	34.1800
epoch	3.0000

Hyperparameters

• Batch Size: 8
• 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.'))