Text this: Causality-aware graph neural networks for functional stratification and phenotype prediction at scale