Předpověď míry opětovného zatčení

Vytvořil/a jsi regresní model přežití využívající kovariáty o každém propuštěném vězni, jako je rodinný stav, počet předchozích zatčení a další. Deset nových vězňů se chystá na propuštění a ty pomocí tohoto modelu předpovíš jejich střední dobu do opětovného zatčení.

Nový DataFrame se jmenuje prison_new a má stejné sloupce jako původní DataFrame prison.

Třída WeibullAFTFitter je naimportována a dříve natrénovaný model je načtený jako aft. Knihovny pandas a numpy jsou importovány jako pd a np.