Table 4 Search space for hyperparameters of GNN models.
From: Cell graph neural networks enable the precise prediction of patient survival in gastric cancer
Hyperparameter | Searching space |
|---|---|
Learning rate | 10−4, 5 × 10−4, 10−3 |
Weight decay (L2) | 10−4, 5 × 10−4, 10−3 |
Hidden units | 256, 512 |
Pooling ratio | 0.5, 0.65, 0.75 |
