IsGarrido/gender_classifier_multi_xlmroberta_base

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

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

Performance Metrics

Metric	Value
loss	0.3350
accuracy	0.9124
f1	0.9120
precision	0.9129
recall	0.9125
runtime	6.9012
samples_per_second	1738.8310
steps_per_second	54.3380
epoch	3.0000

Hyperparameters

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