File size: 690 Bytes
6a8c42f
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
{
  "base_model": "microsoft/deberta-v3-small",
  "mode": "binary",
  "epochs": 5,
  "learning_rate": 2e-05,
  "batch_size": 32,
  "gradient_accumulation_steps": 1,
  "warmup_ratio": 0.1,
  "weight_decay": 0.01,
  "max_length": 256,
  "label_smoothing": 0.15,
  "class_weights": [
    0.9379903674125671,
    1.062009572982788
  ],
  "seed": 42,
  "train_size": 3951,
  "val_size": 1909,
  "test_size": 2140,
  "train_label_dist": {
    "NOT_ACTION": 2098,
    "ACTION": 1853
  },
  "test_f1_macro": 0.8683538522408523,
  "test_f1_action": 0.853220696937698,
  "test_accuracy": 0.8700934579439252,
  "num_labels": 2,
  "label_map": {
    "NEUTRAL": 0,
    "POLICY": 0,
    "ACTION": 1
  }
}