Python에서의 Survival Analysis
Shae Wang
Senior Data Scientist
서로 다른 사건 결과의 확률을 설명하는 수학적 함수입니다.

서로 다른 사건 결과의 확률을 설명하는 수학적 함수입니다.

연속형 확률분포로, 사건까지의 시간을 잘 모델링합니다(원래는 입자 크기 분포에 적용).
$$f(x;\lambda,k)=\frac{k}{\lambda}\bigg(\frac{x}{\lambda}\bigg)^{k-1}e^{-(x/\lambda)^k}$$ $$x\geq0,k>0,\lambda>0$$
모양을 결정

스케일을 결정

한 기업이 고장에 취약한 기계들을 운영합니다...

한 기업이 고장에 취약한 기계들을 운영합니다...

$$f(x;\lambda,k)=\frac{k}{\lambda}\bigg(\frac{x}{\lambda}\bigg)^{k-1}e^{-(x/\lambda)^k} \quad\rightarrow\quad\qquad\qquad S(t)=e^{-(t/\lambda)^\rho}$$

$\rho$는 k와 동일함
$$f(x;\lambda,k)=\frac{k}{\lambda}\bigg(\frac{x}{\lambda}\bigg)^{k-1}e^{-(x/\lambda)^k} \quad\rightarrow\quad f(x;\lambda,k=3)=\frac{3}{\lambda}\bigg(\frac{x}{\lambda}\bigg)^2e^{-(x/\lambda)^3}$$



WeibullFitter 클래스 임포트from lifelines import WeibullFitter
WeibullFitter 인스턴스 생성wb = WeibullFitter()
.fit()으로 데이터에 적합wb.fit(durations, event_observed)
.survival_function_, .lambda_, .rho_, .summary, .predict() 사용DataFrame 이름: mortgage_df
| id | duration | paid_off |
|---|---|---|
| 1 | 25 | 0 |
| 2 | 17 | 1 |
| 3 | 5 | 0 |
| ... | ... | ... |
| 1000 | 30 | 1 |
from lifelines import WeibullFitter
wb = WeibullFitter()
wb.fit(durations=mortgage_df["duration"],
event_observed=mortgage_df["paid_off"])
wb.survival_function_.plot()
plt.show()

print(wb.lambda_, wb.rho_)
6.11 0.94
print(wb.predict(20))
0.05
Python에서의 Survival Analysis