異常：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 ）