MatteoWood's picture
Update README.md
baf9b62
metadata
tags:
  - text-classification
  - sexism
widget:
  - text: I love women
  - text: I hate women
datasets:
  - MatteoWood/bert-sexism-classifier-dataset

Model Trained Using AutoTrain

  • Problem type: Text Classification

Training parameters

  • lr: 5e-05
  • epochs: 5
  • batch size: 8

Validation Metrics

loss: 0.36799120903015137

f1: 0.7044155844155844

precision: 0.7251336898395722

recall: 0.6848484848484848

auc: 0.8958072418537535

accuracy: 0.85775