Downloads · 30 days
13
25% of all-time downloads
IsGarrido/gender_classifier_multi_xlmroberta_base
gender_classifier_multi_xlmroberta_base is a text classification model from IsGarrido. Use it when you need a label for a piece of text. It is set up for transformers.
This model was fine-tuned to classify text into: male, female, neutral
Downloads · 30 days
13
25% of all-time downloads
All-time downloads
51
Public
Parameters
278M
1.1 GB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors1.1 GB · 98%
From the Hugging Face model README
This model was fine-tuned to classify text into: male, female, neutral
| 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 |
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.'))