found: org.apache.spark.sql.Dataset[(Double, Double)] required: org.apache.spark.rdd.RDD[(Double, Double)]
我收到以下错误
1
2
3
| found : org. apache. spark. sql. Dataset[(Double, Double )]
required : org. apache. spark. rdd. RDD[(Double, Double )]
val testMetrics = new BinaryClassificationMetrics (testScoreAndLabel ) |
关于以下代码：
1
2
3
4
| val testScoreAndLabel = testResults.
select("Label", "ModelProbability").
map{ case Row (l :Double,p :Vector ) => (p (1),l ) }
val testMetrics = new BinaryClassificationMetrics (testScoreAndLabel ) |
从错误看来，testScoreAndLabel 是 sql.Dataset 类型，但 BinaryClassificationMetrics 需要 RDD。
如何将 sql.Dataset 转换为 RDD？
我会做这样的事情
1
2
3
| val testScoreAndLabel = testResults.
select("Label", "ModelProbability").
map{ case Row (l :Double,p :Vector ) => (p (1),l ) } |
现在只需执行 testScoreAndLabel.rdd
即可将 testScoreAndLabel 转换为 RDD
1
| val testMetrics = new BinaryClassificationMetrics (testScoreAndLabel. rdd) |
API 文档