【发布时间】:2020-03-07 11:44:39
【问题描述】:
我想使用排名指标计算recall@25 和F1-score@25。
val predictionAndLabels = predicted.
select($"prediction",$"label")
.as[(Double, Double)]
val Arr = predictionAndLabels.rdd.map(x => (Array(x._1),Array(x._2)))
val matrix = new RankingMetrics(Arr)
Array(1, 25).foreach { k =>
println(s"Recall at $k = ${matrix.recallAt(k)}")
}
Array(1, 25).foreach { k =>
println(s"F1-score at $k = ${matrix.F1scoreAt(k)}")
}
错误:
值recallAt 不是 org.apache.spark.mllib.evaluation.RankingMetrics[双]
如何解决?
【问题讨论】:
-
你使用的 spark 版本是什么?
-
我使用的是 spark 2.0.2 版。
标签: scala apache-spark recommendation-engine