Ať žije demokracie!

Dříve v této kapitole jsme prozkoumali a potvrdili problém cenzurování dat v datasetu délky trvání režimů. Teď je čas sestavit svoji první křivku přežití pro odhad průměrné doby trvání.

Tvým úkolem je odhadnout funkci přežití a vykreslit křivku přežití zobrazující pravděpodobnosti přežití. K odhadu funkce přežití vytvoříš instanci objektu KaplanMeierFitter a natrénuješ ho na příslušných sloupcích dat. Názvy parametrů jsou pro tebe předvyplněny – více o trénování funkcí přežití se dozvíš v dalších kapitolách.

Balíček pandas je načten jako pd, modul matplotlib.pyplot jako plt a třída KaplanMeierFitter je importována z lifelines. DataFrame s délkami trvání je načten a uložen jako regime_durations.