Table 1.
Graph convolution model hyperparameters
Group | Hyperparameter | Default value |
---|---|---|
Input | Maximum number of atoms per molecule | 60 |
Maximum atom pair graph distance | 2 | |
Weave | Number of Weave modules | 1 |
(A → A)0 convolution depth | 50 | |
(A → P)0 convolution depth | 50 | |
(P → P)0 convolution depth | 50 | |
(P → A)0 convolution depth | 50 | |
(A → A)1 convolution depth | 50 | |
(P → P)1 convolution depth | 50 | |
Reduction | Final atom layer convolution depth | 128 |
Reduction to molecule-level features | Gaussian histogram | |
Post-reduction | Fully-connected layers (number of units per layer) | 2000, 100 |
Training | Batch size | 96 |
Learning rate | 0.003 | |
Optimization method | Adagrad |