File size: 441 Bytes
e0c1384 d30bee5 e0c1384 5c38da3 d30bee5 |
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 |
model:
d_model: 256
d_state: 16
d_conv: 4
expand: 2
n_layers: 4
dropout: 0.1
data:
batch_size: 32
max_nodes: 10000
test_split: 0.2
training:
learning_rate: 0.01
weight_decay: 0.0005
epochs: 200
patience: 20
warmup_epochs: 10
min_lr: 1e-6
ordering:
strategy: "bfs" # bfs, spectral, degree, community
preserve_locality: true
evaluation:
metrics: ["accuracy", "f1_macro", "f1_micro"]
save_best: true |