Table 2.
Hyperparameters | Values | Average accuracy | AUROC |
---|---|---|---|
Batch size | 1 | 77.15%a | 0.8246a |
8 | 75.85%a | 0.8009a | |
16 | 75.85%a | 0.8183a | |
32 | 76.32%a | 0.8118a | |
64 | 76.21%a | 0.8212a | |
128 | 76.09%a | 0.8249a | |
256 | 75.62%a | 0.8246a | |
Number of GCN layers | one | 75.38%a | 0.7850a |
two | 77.15%a | 0.8246a | |
three | 76.68%a | 0.8201a | |
Pocket size | 5 Å res around ligand | 77.15%a | 0.8246a |
10 Å res around ligand | 76.33%a | 0.8114a | |
15 Å res around ligand | 73.73%a | 0.8010a | |
All protein | 73.97%a | 0.7780a | |
Pooling layer | Max pooling | 77.15%a | 0.8246a |
Mean pooling | 72.44%a | 0.7692a | |
Sum pooling | 72.20%a | 0.7733a | |
Inclusion of bond type encoding | No | 77.15%a | 0.8246a |
Yes | 77.97%a | 0.8278a | |
Optimized hyperparameters | use | 77.46%b | 0.8531b |
Note: a metrics value on validation set, b metrics value on test set; bold means optimal parameters.