The example code of time-dependent Cox Survival Neural Network (tdCoxSNN)
-
Data contains the cleaned PBC2 data
-
funcs_util contains the functions for calculating baseline hazards and the dynamic survival probability
-
PyTorch contains the loss function in PyTorch version and PyTorch example code
-
Tensorflow contains the loss function in Tensorflow version and Tensorflow example code
-
rTensorflow contains the loss function in R-Tensorflow version and R-Tensorflow example code
The arXiv version is available. An earlier version has won the 2023 JSM Risk Analysis Section Student Paper Award.