Community Browser
cancel
Showing results for
Did you mean:
DataRobot Community is now read-only until August 9th, at which time it will be deactivated. Please visit docs.datarobot.com for product documentation or reach out to support@datarobot.com with any specific product-related questions.

Survival Analysis

Does DataRobot include any models for survival analysis?

1 Solution

Accepted Solutions
DataRobot Alumni

DataRobot does not have the traditional survival models like Cox Proportional Hazards model. But, you can convert the problem as a binary classification problem, which can be done using DataRobot. This can be done by breaking down the "patient" level data into "patient-time" format by creating multiple rows for the patient at discrete time intervals. The target for the binary classification model would be if the patient survives a particular time interval. For uncensored patients, the target would be 0 for all the time intervals except the last time interval while for censored patients, the target would be 0 for all the time intervals. The predictions from the model would indicate if the probability of the event happening a particular time interval which would be instantaneous hazard probability. Instantaneous survival probability would be complement of instantaneous hazard probability and you can recreate survival curves by cumulating the instantaneous survival probability.

DataRobot Alumni

DataRobot does not have the traditional survival models like Cox Proportional Hazards model. But, you can convert the problem as a binary classification problem, which can be done using DataRobot. This can be done by breaking down the "patient" level data into "patient-time" format by creating multiple rows for the patient at discrete time intervals. The target for the binary classification model would be if the patient survives a particular time interval. For uncensored patients, the target would be 0 for all the time intervals except the last time interval while for censored patients, the target would be 0 for all the time intervals. The predictions from the model would indicate if the probability of the event happening a particular time interval which would be instantaneous hazard probability. Instantaneous survival probability would be complement of instantaneous hazard probability and you can recreate survival curves by cumulating the instantaneous survival probability.

Latest Topics