CamDavidsonPilon / lifelines

Survival analysis in Python
lifelines.readthedocs.org
MIT License
2.32k stars 551 forks source link

stratification in CoxTimeVaryingFitter #1614

Open andreatrivellahappymoney opened 1 month ago

andreatrivellahappymoney commented 1 month ago

when fitting a CoxTimeVaryingFitter with stratification, the attribute baselinesurvival of the fitted model does not distinguish between different strata. It would be very helpful to be able to access the underlying baselinesurvival for each strata (the same problem happens for baseline_cumulativehazard and baselinehazard)