Compare Weibull model parameters
Some convicts in the study were released on parole. Because parolees experience stricter restrictions, you suspect that their initial arrest rate is high but decreases as time goes on, while other convicts' arrest rate increases as time goes on.
You will fit a Weibull estimator to parolees and non-parolees separately. You will then compare their model parameters to verify your hypothesis. The DataFrame you will use is called prison
.
The matplotlib.pyplot
library is imported as plt
, the pandas
and numpy
libraries are imported as pd
and np
, respectively. The WeibullFitter
class has been imported for you and a new instance called wb
has been created. Use the console to explore the DataFrame as needed.
This exercise is part of the course
Survival Analysis in Python
Hands-on interactive exercise
Have a go at this exercise by completing this sample code.
# Mask for parole
parole = (____ == ____)