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.

Cet exercice fait partie du cours

Survival Analysis in Python

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

Essayez cet exercice en complétant cet exemple de code.

# Check PH assumption
print(cph.____(training_df=____, p_value_threshold=____))