org.apache.spark.mllib.linalg.Vector到DataFrame標量

Question

我有如下輸出

val output = Vectors.dense(Vectors.dense(pred.toDouble).toArray ++ predProb.toArray)
output: org.apache.spark.mllib.linalg.Vector = [0.0,0.9279938116118619,0.07200618838813808]

我想將其轉換為具有以下作為列值的DataFrame

case class Prob(Pred: Double, NoProb: Double, YesProb: Double)

有人可以幫忙嗎？

謝謝

巴拉

Answer 1

val df = output.map（_。toArray）.map {case Array（p0，p1，p2）=> Prob（p0，p1，p2）}。toDF（）