XGBoost 生存模型预测

Question

Xgboost 的文档暗示使用 Cox PH 损失训练的模型的输出将是个人预测乘数（相对于基线风险）的幂。有没有办法从这个模型中提取基线风险来预测每个人的整个生存曲线？

survival:cox：右删失生存时间数据的 Cox 回归 （负值被认为是右删失）。注意预测 以风险比量表返回（即 HR = 比例风险函数中的 exp(marginal_prediction) h(t) = h0(t) * HR)

Answer 1

不，我想不是。解决方法是将基线危险纳入另一个包中，例如

from sksurv.linear_model import CoxPHSurvivalAnalysis

或用 R 表示

require(survival)

。然后，您可以使用 XGBoost 的预测输出作为拟合基线的乘数。请记住，如果基线采用对数刻度，则使用

output_margin=True

并添加预测。