name 'survival_probability' is not defined

如果您遇到了 `name 'survival_probability' is not defined` 的错误，这可能是因为您没有正确导入 `survival_probability` 函数。请确保您已经正确导入了 `lifelines.utils` 中的 `survival_probability` 函数。

以下是一个完整的示例代码，其中包括正确的导入语句：

from lifelines import CoxPHFitter
from lifelines.datasets import load_rossi
from lifelines.utils import survival_probability
from sklearn.metrics import roc_auc_score

# 加载罗西数据集
rossi_dataset = load_rossi()

# 拟合COX模型
cph = CoxPHFitter()
cph.fit(rossi_dataset, duration_col='week', event_col='arrest')

# 计算生存概率
survival_prob = survival_probability(cph, rossi_dataset)

# 计算AUC
auc = roc_auc_score(rossi_dataset['arrest'], survival_prob)

在这个示例代码中，我们使用 `lifelines.utils` 中的 `survival_probability` 函数计算生存概率，并使用 `sklearn.metrics` 中的 `roc_auc_score` 函数计算AUC。请注意，在导入语句中，我们确保正确导入了 `lifelines.utils` 中的 `survival_probability` 函数。