如何在Pyspark的回归任务中获取标签的概率

Question

我想知道如何使用PySpark ML库在回归任务期间获得每个标签的概率。

使用GBTRegressor()时，我有三种不同的标签：

GBTR_model.transform(test_dataset)

我只获得有关最常见标签的概率，而我没有获得其余标签的其他两个概率。