UDHOV commited on
Commit
97189b6
·
verified ·
1 Parent(s): 862f41e

Add training configuration

Browse files
Files changed (1) hide show
  1. training_config.json +9 -9
training_config.json CHANGED
@@ -5,7 +5,7 @@
5
  "batch_size": 8,
6
  "gradient_accumulation_steps": 4,
7
  "learning_rate": 1e-05,
8
- "num_epochs": 7,
9
  "max_length": 256,
10
  "class_weights": [
11
  0.5,
@@ -22,15 +22,15 @@
22
  "test_metrics": {
23
  "model": "XLM-RoBERTa-large",
24
  "strategy": "standard",
25
- "test_accuracy": 0.7386206896551725,
26
- "test_precision": 0.5264896402013559,
27
- "test_recall": 0.6153446388098774,
28
- "test_f1": 0.5588019554444394,
29
  "per_class_f1": {
30
- "NO": 0.8090539756239118,
31
- "OO": 0.6799999999999999,
32
- "OR": 0.4,
33
- "OS": 0.34615384615384615
34
  }
35
  }
36
  }
 
5
  "batch_size": 8,
6
  "gradient_accumulation_steps": 4,
7
  "learning_rate": 1e-05,
8
+ "num_epochs": 6,
9
  "max_length": 256,
10
  "class_weights": [
11
  0.5,
 
22
  "test_metrics": {
23
  "model": "XLM-RoBERTa-large",
24
  "strategy": "standard",
25
+ "test_accuracy": 0.6951724137931035,
26
+ "test_precision": 0.4887482661901078,
27
+ "test_recall": 0.5948218174128002,
28
+ "test_f1": 0.5212540822752333,
29
  "per_class_f1": {
30
+ "NO": 0.7771295215869312,
31
+ "OO": 0.6164658634538153,
32
+ "OR": 0.42276422764227645,
33
+ "OS": 0.2686567164179105
34
  }
35
  }
36
  }