In a previous post we covered the assumptions of the Cox Proportional Hazards model. The model places very little restriction on the shape of the baseline hazard function or on the linear predictor, however, it does assume that the linear predictor has a constant proportional effect on the hazard function.
This assumption is definitely non-trivial, and can be prone to fail. An interesting thing to note is that this assumptions can fail even if every member of the population actually meets proportional hazard individually. For example, if there is heterogenity in the population (some people have shorter lives than others), then through time we will be left with the longer lived members of the population, so the measured effect of beta will change through time.
The most intuitive way to test the assumption is to just include time varying covariates in the regressions and see whether they have siginificant coefficients. If they do, then there is time varying information that we are not capturing and hence the assumption is false.
Unfortunately, there isn’t an easy way to do this in python, as lifelines doesn’t offer time varying covariates.
More information on how to test the assumptions can be found at the following links: