distilbert-bias-classifier / bias_label_config.json
guisalim's picture
Bias classifier — local Mac run | F1 macro: 0.2678 | device: mps
2c167f6 verified
{
"labels": [
"racial",
"gender",
"religious",
"age",
"sexual_orientation"
],
"id2label": {
"0": "racial",
"1": "gender",
"2": "religious",
"3": "age",
"4": "sexual_orientation"
},
"label2id": {
"racial": 0,
"gender": 1,
"religious": 2,
"age": 3,
"sexual_orientation": 4
},
"threshold": 0.4,
"device": "mps",
"model": "distilbert-base-uncased",
"task": "multi_label_classification"
}