Key Hyperparameters for the GNN Module
The edge threshold similarity parameter indicates the minimum similarity score required to establish an edge between nodes in the graph. Note that the final hyperparameter values were determined after multiple Monte Carlo empirical validations had been conducted on a subset of the training data sets.
| Parameter | Value |
|---|---|
| Hidden_dimension | 128 |
| No. of nodes | 20 |
| No. of epochs | 11 |
| No. of layers | 8 |
| Optimizer | Adam |
| Learning rate | 0.001 |
| Weight decay | 5e – 4 |
| Loss function | MSE |
| Minimum value for edge threshold similarity | 0.5 |