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

Model Trained Using AutoTrain

  • Problem type: Text Classification

Training parameters

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

Validation Metrics

loss: 0.40127208828926086

f1: 0.6585251291686237

precision: 0.6154521510096576

recall: 0.7080808080808081

auc: 0.8725866639820129

accuracy: 0.81825