gpt4 book ai didi

scala - 如何拆分 ML 管道逻辑回归产生的预测概率

转载 作者:行者123 更新时间:2023-12-01 11:30:23 25 4
gpt4 key购买 nike

我正在尝试使用 ML 管道和 DataFrame API 从逻辑模型中提取预测概率。预测概率的输出是一个列向量,它存储每个类 (0, 1) 的预测概率,如下所示。我想知道如何只提取第 1 类的概率。谢谢!

prob
"[0.13293408418007766,0.8670659158199223]"
"[0.1335112097146626,0.8664887902853374]"

最佳答案

像这样的 UDF 应该可以工作:

import org.apache.spark.sql.functions.udf

val getPOne = udf((v: org.apache.spark.mllib.linalg.Vector) => v(1))
model.transform(testDf).select(getPOne($"probability"))

关于scala - 如何拆分 ML 管道逻辑回归产生的预测概率,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/32570799/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com