jaredleekatzman / DeepSurv

DeepSurv is a deep learning approach to survival analysis.
MIT License
566 stars 166 forks source link

output risk for computing timeROC #84

Open EQUIWDH opened 11 months ago

EQUIWDH commented 11 months ago

Hello, I am considering about increasing evaluation for deepsurv. I am wondering it's reasonable that use output risk $h_{\theta}(x)$ to calculate survival lines just imitating breslow and KM estimation method because they come from full likelihood of CoxPH. such the base line cumulative hazard function $$\Lambda0 (t) = \sum{i=1}^{N} \frac{\deltai}{ \sum{j \in R( T{(i)} ) } \exp(h{\theta}(x))} I( t \geq T_{(i)} )$$ and then, calculate timeROC. Thanks!