在 pyspark LogisticRegression 输出中，Whick 概率属于哪个类？

Question

我的代码如下：

from pyspark.ml.classification import LogisticRegression
lr = LogisticRegression(featuresCol="features", labelCol="label")
lrm = lr.fit(transformed)
predictions = lrm.transform(transformed)
predictions.select('probability')

这给了我 2 个概率，因为我有 2 个班级。 但我不知道哪个概率属于哪个类别。如何找到那个。谢谢

Answer 1

spark 预测基于 p>0.5 的类别。所以从预测数据框中选择概率和预测列。检查数据帧的第一行，查看哪个元素的值高于 0.5 并查看预测类。因此，例如元素 1 的值为 0.56，预测为 2 类。那么概率的第 0 个元素属于 1 类概率，第 1 个元素是 2 类。