Update config.yaml
Browse files- config.yaml +11 -5
config.yaml
CHANGED
@@ -12,11 +12,17 @@ data:
|
|
12 |
test_split: 0.2
|
13 |
|
14 |
training:
|
15 |
-
learning_rate:
|
16 |
-
weight_decay: 0.
|
17 |
-
epochs:
|
18 |
-
patience:
|
|
|
|
|
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
|