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.4.9.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | fbcd848ccaef3f91efc2487db7299080576954184198d4e5eea9ad48fdc73277 |
|
MD5 | e8f080880b7023bde6be64cb1289b81d |
|
BLAKE2b-256 | 85a4efc367e17ccb8ec6445ffbc8032ef1550669f41d543b02917ddb3ead9b07 |
Close
Hashes for survival_data_handler-2024.4.9-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 06d83cff76c16d8cdb827cfa7c35173c7f875ba77cc4e745631108963d91bdb3 |
|
MD5 | c7dd17818f5517330b4bc78eb9870512 |
|
BLAKE2b-256 | fdcb9d22354bfac54619fd10f3fce56ee3250953cabd151afe7223df7b328753 |