gender-classifier
Collection
7 items • Updated
Gender Classifier (Fine-tuned PeterPanecillo/PlanTL-GOB-ES-roberta-base-bne-copy)
This model was fine-tuned to classify text into: male, female, neutral
Performance Metrics
| Metric | Value |
|---|---|
| loss | 0.4235 |
| accuracy | 0.9050 |
| f1 | 0.9050 |
| precision | 0.9050 |
| recall | 0.9050 |
| runtime | 2.1530 |
| samples_per_second | 2786.8420 |
| steps_per_second | 43.6610 |
| epoch | 3.0000 |
Hyperparameters
- Batch Size: 64
- 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.'))
- Downloads last month
- 35
Model tree for IsGarrido/gender_classifier_es_roberta_base
Base model
PlanTL-GOB-ES/roberta-base-bne