-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathpubmed_gnns.log
41 lines (27 loc) · 4.62 KB
/
pubmed_gnns.log
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
[Stage 1736:> (0 + 1) / 1]/home/dwalke/.local/lib/python3.10/site-packages/transformers/utils/generic.py:441: UserWarning: torch.utils._pytree._register_pytree_node is deprecated. Please use torch.utils._pytree.register_pytree_node instead.
_torch_pytree._register_pytree_node(
[Stage 1737:> (0 + 1) / 1]/home/dwalke/.local/lib/python3.10/site-packages/transformers/utils/generic.py:441: UserWarning: torch.utils._pytree._register_pytree_node is deprecated. Please use torch.utils._pytree.register_pytree_node instead.
_torch_pytree._register_pytree_node(
[Stage 1738:> (0 + 1) / 1]/home/dwalke/.local/lib/python3.10/site-packages/transformers/utils/generic.py:441: UserWarning: torch.utils._pytree._register_pytree_node is deprecated. Please use torch.utils._pytree.register_pytree_node instead.
_torch_pytree._register_pytree_node(
[Stage 1739:> (0 + 1) / 1]/home/dwalke/.local/lib/python3.10/site-packages/transformers/utils/generic.py:441: UserWarning: torch.utils._pytree._register_pytree_node is deprecated. Please use torch.utils._pytree.register_pytree_node instead.
_torch_pytree._register_pytree_node(
Total Trials: 160: 160 succeeded, 0 failed, 0 cancelled.
[2K0it [00:00, ?it/s]
[2K[2KAccuracy for GCN: 0.8709243337313334 +- 0.01117211026289771
Accuracy for GAT: 0.8610334992408752 +- 0.011076087305362884
Accuracy for Cheb: 0.8872039318084717 +- 0.0021130316584040136
Accuracy for SAGE: 0.8872036933898926 +- 0.0041809939156471545
{'GCN': array([0.85531723, 0.87659764, 0.88085812]), 'GAT': array([0.8629241 , 0.84662199, 0.87355441]), 'Cheb': array([0.88711399, 0.88983566, 0.88466215]), 'SAGE': array([0.89167809, 0.88161898, 0.88831401])}
{'GCN': [{'add_self_loops': True, 'dropout': 0.41455320564578013, 'hidden_dim': 304.0, 'lr': 0.06446442154949868, 'normalize': True, 'out_dim': 3, 'weight_decay': 3.540907311149174e-05}, {'add_self_loops': True, 'dropout': 0.673671485815347, 'hidden_dim': 160.0, 'lr': 0.04725158965214731, 'normalize': True, 'out_dim': 3, 'weight_decay': 5.045576594174209e-05}, {'add_self_loops': True, 'dropout': 0.7700318963951692, 'hidden_dim': 144.0, 'lr': 0.09826584935460114, 'normalize': True, 'out_dim': 3, 'weight_decay': 1.7557663887622646e-05}], 'GAT': [{'add_self_loops': True, 'dropout': 0.6609137272157298, 'heads': 8.0, 'hidden_dim': 480.0, 'lr': 0.010086425183218072, 'out_dim': 3, 'weight_decay': 2.4131050167233953e-05}, {'add_self_loops': True, 'dropout': 0.5973554924182691, 'heads': 6.0, 'hidden_dim': 48.0, 'lr': 0.060061607442611874, 'out_dim': 3, 'weight_decay': 1.208681792755945e-05}, {'add_self_loops': True, 'dropout': 0.022269667956317546, 'heads': 4.0, 'hidden_dim': 16.0, 'lr': 0.06076218610684613, 'out_dim': 3, 'weight_decay': 1.9360996329385315e-05}], 'Cheb': [{'K': 4.0, 'dropout': 0.2466960684873182, 'hidden_dim': 112.0, 'lr': 0.05025518418822697, 'normalization': 'sym', 'out_dim': 3, 'weight_decay': 8.206388404941222e-05}, {'K': 2.0, 'dropout': 0.27417556777463925, 'hidden_dim': 16.0, 'lr': 0.039440756352912706, 'normalization': 'sym', 'out_dim': 3, 'weight_decay': 0.00015337342485399427}, {'K': 2.0, 'dropout': 0.6862435105644612, 'hidden_dim': 176.0, 'lr': 0.033949396260314714, 'normalization': 'sym', 'out_dim': 3, 'weight_decay': 7.070504910318309e-05}], 'SAGE': [{'dropout': 0.7562954407858262, 'hidden_dim': 608.0, 'lr': 0.010335624071294208, 'normalize': False, 'out_dim': 3, 'project': False, 'root_weight': True, 'weight_decay': 0.00010189916788041466}, {'dropout': 0.7494470187360279, 'hidden_dim': 352.0, 'lr': 0.0010972739812105494, 'normalize': True, 'out_dim': 3, 'project': False, 'root_weight': True, 'weight_decay': 0.00627483471953386}, {'dropout': 0.7390447015065373, 'hidden_dim': 176.0, 'lr': 0.025062368722206817, 'normalize': False, 'out_dim': 3, 'project': False, 'root_weight': True, 'weight_decay': 4.6774675841434645e-05}]}
Train time for GCN: 1.1173648834228516 +- 0.3729569555417256
Train time for GAT: 2.330679972966512 +- 1.3577021572717933
Train time for Cheb: 1.5806047916412354 +- 0.18372708646430375
Train time for SAGE: 2.350901444753011 +- 0.35377398125670484