File size: 1,041 Bytes
219d96c
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
{
  "architectures": [
    "MultiLabelIntentClassifier"
  ],
  "model_type": "distilbert",
  "num_labels": 6,
  "id2label": {
    "0": "trolling",
    "1": "dismissive",
    "2": "manipulative",
    "3": "emotionally_reactive",
    "4": "constructive",
    "5": "unclear"
  },
  "label2id": {
    "trolling": 0,
    "dismissive": 1,
    "manipulative": 2,
    "emotionally_reactive": 3,
    "constructive": 4,
    "unclear": 5
  },
  "base_model": "distilbert-base-uncased",
  "task_specific_params": {
    "text-classification": {
      "problem_type": "multi_label_classification",
      "num_labels": 6
    }
  },
  "intent_categories": [
    "trolling",
    "dismissive",
    "manipulative",
    "emotionally_reactive",
    "constructive",
    "unclear"
  ],
  "model_architecture": "MultiLabelIntentClassifier",
  "training_metrics": {
    "f1_macro": 0.77,
    "f1_trolling": 0.943,
    "f1_dismissive": 0.85,
    "f1_manipulative": 0.907,
    "f1_emotionally_reactive": 0.939,
    "f1_constructive": 0.989,
    "f1_unclear": 0.0
  }
}