Extract survival curves

[Justinewits]

Hello @jaredleekatzman , Thank you for the DeepSurv algorithm, can you please advise on how one can extract the survival curves from the implemented algorithm on their dataset? Thank you!