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
|