请教大神，关于预后预测模型风险分层的疑问

数据分成训练集和验证集，训练集经过单、多因素COX回归，建立预测模型。现在需要通过预测模型预测的中位数对数据进行风险分层。模型在训练集预测的中位数A，验证集预测的中位数B，

对于验证集的分层，是用A 去把验证集分层高风险和低风险组，还是用B去分组啊？很疑惑， 请大神三言两语解答。