causallib.survival.marginal_survival module
- class causallib.survival.marginal_survival.MarginalSurvival(survival_model: Optional[Any] = None)[source]
Bases:
causallib.survival.weighted_survival.WeightedSurvival
Marginal (un-adjusted) survival estimator. Essentially it is a degenerated WeightedSurvival instance without a weight model.
Marginal (un-adjusted) survival estimator. :param survival_model: Three alternatives:
None - compute non-parametric KaplanMeier survival curve
Scikit-Learn estimator (needs to implement predict_proba) - compute parametric curve by fitting a time-varying hazards model
lifelines UnivariateFitter - use lifelines fitter to compute survival curves from events and durations