No project description provided
Project description
Survival data handler
import pandas as pd
from lifelines import CoxPHFitter
from lifelines.datasets import load_rossi
rossi = load_rossi()
cph = CoxPHFitter()
cph.fit(rossi, duration_col='week', event_col='arrest')
curves = cph.predict_survival_function(rossi).T
curves.columns = pd.to_timedelta(curves.columns.to_numpy() * 7, unit="D")
print(curves.head())
7 days 00:00:00 | 14 days 00:00:00 | 21 days 00:00:00 | 28 days 00:00:00 | 35 days 00:00:00 | 42 days 00:00:00 | 49 days 00:00:00 | 56 days 00:00:00 | 63 days 00:00:00 | 70 days 00:00:00 | |
---|---|---|---|---|---|---|---|---|---|---|
0 | 0.997616 | 0.99523 | 0.992848 | 0.990468 | 0.988085 | 0.985699 | 0.983305 | 0.971402 | 0.966614 | 0.964223 |
1 | 0.993695 | 0.987411 | 0.981162 | 0.974941 | 0.968739 | 0.962552 | 0.95637 | 0.926001 | 0.913958 | 0.907978 |
2 | 0.994083 | 0.988183 | 0.982314 | 0.976468 | 0.970639 | 0.96482 | 0.959004 | 0.930402 | 0.919043 | 0.913399 |
3 | 0.999045 | 0.998089 | 0.997133 | 0.996176 | 0.995216 | 0.994254 | 0.993287 | 0.98846 | 0.986508 | 0.985531 |
4 | 0.997626 | 0.99525 | 0.992878 | 0.990507 | 0.988135 | 0.985758 | 0.983374 | 0.97152 | 0.966752 | 0.96437 |
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Close
Hashes for survival_data_handler-2024.1.1.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | e947f77dbecbbb6e9fec6aaab1cc3ddbe74c12ff6097d16e6422dab0dd2c6c7f |
|
MD5 | 2f525e44e7464406876bf4253acbb54e |
|
BLAKE2b-256 | 9bc34c2b0c0c021bcbb223785eb53237804a8e21e2c9a3ad5514c731aa89233b |
Close
Hashes for survival_data_handler-2024.1.1-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 68bac256b1b1cc08e83586523390a0f2f9e4f0dbfdbe84f86722d557780e3a56 |
|
MD5 | 733835733dae0cee1db0df46445d47bc |
|
BLAKE2b-256 | bb1173d97079cdf2d696a301df1c91dec77a149d502647203e747495bb856927 |