| { | |
| "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" | |
| } |