fit_survival_model function

Fit a survival sub-model