首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >Spark :回归模型阈值和精度

Spark :回归模型阈值和精度
EN

Stack Overflow用户
提问于 2015-12-11 13:04:56
回答 3查看 2.8K关注 0票数 4

我有逻辑回归模式,其中我显式地将阈值设置为0.5。

代码语言:javascript
运行
复制
model.setThreshold(0.5)

我训练模型,然后我想获得基本的统计数据--精度、召回率等。

这是我在评估模型时所做的:

代码语言:javascript
运行
复制
val metrics = new BinaryClassificationMetrics(predictionAndLabels)

val precision = metrics.precisionByThreshold


precision.foreach { case (t, p) =>

      println(s"Threshold is: $t, Precision is: $p")

    }

我得到的结果只有0.0和1.0作为阈值,0.5被完全忽略。

下面是上面循环的输出:

阈值为: 1.0,精度为: 0.8571428571428571

阈值为: 0.0,精度为: 0.3005181347150259

当我调用metrics.thresholds()时,它也只返回两个值: 0.0和1.0。

如何获取阈值为0.5的查准率和召回值?

EN

Stack Overflow用户

发布于 2016-03-17 22:39:43

在进行预测之前,您需要清除模型阈值。清除阈值使您的预测返回一个分数,而不是分类标签。否则,您将只有两个阈值,即您的标签0.0和1.0。

代码语言:javascript
运行
复制
model.clearThreshold()

来自predictionsAndLabels的元组应该类似于(0.6753421,1.0),而不是(1.0,1.0)

看一看https://github.com/apache/spark/blob/master/examples/src/main/scala/org/apache/spark/examples/mllib/BinaryClassificationMetricsExample.scala

如果输入很大,您可能仍然希望设置numBins来控制点的数量。

票数 5
EN
查看全部 3 条回答
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/34216481

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档