Skip to content

Instantly share code, notes, and snippets.

@alonsosilvaallende

alonsosilvaallende/PerfectPrediction.py

Last active October 8, 2019 09:58
Show Gist options
  • Save alonsosilvaallende/7897783481744dc243871ae35932f336 to your computer and use it in GitHub Desktop.
Save alonsosilvaallende/7897783481744dc243871ae35932f336 to your computer and use it in GitHub Desktop.
Download ZIP
Perfect prediction
Raw
PerfectPrediction.py
from lifelines.utils import concordance_index
names = ['Alice', 'Bob', 'Carol', 'Dave', 'Eve']
events = [1, 2, 3, 4, 5]
preds = [1, 2, 3, 4, 5]
df = pd.DataFrame(data={'Churn times': events, 'Predictions': preds}, index=names)
print(df)
print(f'Concordance index: {concordance_index(events, preds)}')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment