kfoughali commited on
Commit
d30bee5
·
verified ·
1 Parent(s): f438f4e

Update config.yaml

Browse files
Files changed (1) hide show
  1. config.yaml +11 -5
config.yaml CHANGED
@@ -12,11 +12,17 @@ data:
12
  test_split: 0.2
13
 
14
  training:
15
- learning_rate: 1e-4
16
- weight_decay: 0.01
17
- epochs: 100
18
- patience: 10
 
 
19
 
20
  ordering:
21
  strategy: "bfs" # bfs, spectral, degree, community
22
- preserve_locality: true
 
 
 
 
 
12
  test_split: 0.2
13
 
14
  training:
15
+ learning_rate: 0.01
16
+ weight_decay: 0.0005
17
+ epochs: 200
18
+ patience: 20
19
+ warmup_epochs: 10
20
+ min_lr: 1e-6
21
 
22
  ordering:
23
  strategy: "bfs" # bfs, spectral, degree, community
24
+ preserve_locality: true
25
+
26
+ evaluation:
27
+ metrics: ["accuracy", "f1_macro", "f1_micro"]
28
+ save_best: true