¡Viva la democracia!

Antes en el capítulo, exploramos y confirmamos el problema de censura en el conjunto de datos de duración de regímenes. Ahora, toca construir tu primera curva de supervivencia para estimar la duración media.

Tu tarea es estimar una función de supervivencia y trazar una curva de supervivencia para mostrar las probabilidades de supervivencia. Para estimar la función de supervivencia, crearás una instancia del objeto KaplanMeierFitter y la ajustarás a las columnas de datos adecuadas. Ya tienes los nombres de los parámetros, y aprenderás más sobre cómo ajustar funciones de supervivencia en capítulos posteriores.

El paquete pandas está cargado como pd, el módulo matplotlib.pyplot está cargado como plt, y la clase KaplanMeierFitter está importada desde lifelines. El DataFrame de duraciones está cargado y guardado como regime_durations.