predict_survival function

Predict survival probabilities from a survival model