/GNNs-cox

Primary LanguageJupyter Notebook

GNNs-cox

Time to event prediction using Graph Neural Networs.

This is an extension of the pycox tutorial with CNNs to make it work with graph neural networks. It works with DGL library at the moment. The example uses randomly generated graphs.

The rest of the repo is work in progress, check soon for updates! :D