异常：java.lang.Double 不能转换为 [D

Question

我有逻辑回归模型，我想计算recall@25

val predictionAndLabels = predicted
.select("prediction", "label")
.rdd.map(x => (x(0).asInstanceOf[Array[Double]], x(1)
.asInstanceOf[Array[Double]]))
val matrix = new RankingMetrics(predictionAndLabels)
 Array(1, 25).foreach { k =>
  println(s"Recall at $k = ${matrix.RecallAt(k)}")
}

出现这个异常

java.lang.Double 不能转换为 [D

如何解决？

Answer 1

RankingMetrics 需要 RDD(array, array) ，因为你有 RDD(double,double) 这意味着你正在传递INVALID参数。

一般用于类型转换，使用映射转换

假设 predictionAndLabels.rdd 类型为 RDD(double,double)

pairArrRDD = predictionAndLabels.rdd.map(x => (Array(x._1),Array(x._2))) //这是scala

val 矩阵 = 新排名指标（ pairArrRDD ）