File size: 1,161 Bytes
1256444
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
49
50
51
52
53
54
55
{
  "base_model": "distilbert-base-uncased",
  "n_train": 88,
  "n_eval": 22,
  "eval_accuracy": 0.8636363636363636,
  "eval_f1_macro": 0.7904761904761904,
  "eval_f1_high": 0.9142857142857143,
  "confusion_matrix": [
    [
      3,
      3
    ],
    [
      0,
      16
    ]
  ],
  "baseline_always_predict_high": 0.7272727272727273,
  "baseline_random": 0.5,
  "classification_report": {
    "0": {
      "precision": 1.0,
      "recall": 0.5,
      "f1-score": 0.6666666666666666,
      "support": 6.0
    },
    "1": {
      "precision": 0.8421052631578947,
      "recall": 1.0,
      "f1-score": 0.9142857142857143,
      "support": 16.0
    },
    "accuracy": 0.8636363636363636,
    "macro avg": {
      "precision": 0.9210526315789473,
      "recall": 0.75,
      "f1-score": 0.7904761904761904,
      "support": 22.0
    },
    "weighted avg": {
      "precision": 0.8851674641148325,
      "recall": 0.8636363636363636,
      "f1-score": 0.8467532467532467,
      "support": 22.0
    }
  },
  "training_args": {
    "epochs": 5,
    "batch_size": 8,
    "learning_rate": 3e-05,
    "weight_decay": 0.01,
    "warmup_ratio": 0.1,
    "seed": 42
  }
}