Test the PH assumption automatically

For other continuous variables in prison, you will test the proportional hazards assumption using the .check_assumptions() method.

cph is the fitted Cox PH model using all columns, and it has been loaded.

The CoxPHFitter class has been imported. The pandas and numpy libraries are imported as pd and np, respectively.

100 XP

Use .check_assumptions() at significance level 0.1 to check the proportional hazards assumption for prison.