This code trains a simple graph neural network on bacs/biobank data using pytorch-geometric. Hyperparameters are tuned with optuna. Feature importances can then be assigned with shap.
The code operates on datasets that are proprietary, and so is only shared for illustrative purposes. The main script is xvalidate.py.
Feel free to reuse/repurpose as you see fit.