Gender Classifier (Fine-tuned answerdotai/ModernBERT-base)

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

Performance Metrics

Metric Value
loss 0.2923
accuracy 0.9330
f1 0.9326
precision 0.9332
recall 0.9332
runtime 27.4919
samples_per_second 218.2460
steps_per_second 3.4190
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
9
Safetensors
Model size
0.1B params
Tensor type
F32
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for IsGarrido/gender_classifier_en_modernbert_base

Finetuned
(985)
this model

Collection including IsGarrido/gender_classifier_en_modernbert_base