error: type mismatch; found : org.apache.spark.rdd.RDD[org.apache.spark.mllib.regression.LabeledPoint] required: org.apache.spark.sql.Dataset[_] val lrModel = lr.fit(train)
时间: 2023-09-19 19:05:48 浏览: 152
type mismatch
这个错误是因为你的lr.fit(train)方法需要一个DataFrame类型的数据集,而你传入的是一个RDD类型的数据集。你需要使用SparkSession将RDD转换为DataFrame。例如:
```
import spark.implicits._
val trainDF = train.toDF()
val lrModel = lr.fit(trainDF)
```
这样就可以将RDD转换为DataFrame,然后就可以成功地训练你的模型了。
阅读全文