如何将LogisticRegressionWIthLBFGS与Spark中的Dataframe一起使用

Question

我想在一个旨在使用ML算法和Dataframes的程序中使用MLLIB LogisticRegressionWithLBFGS。 有没有一种简单的方法可以将我的数据帧转换为RDD而无需再次编写新程序？ 这样的事情不起作用：

df3.cache()
val model = new LogisticRegressionWithLBFGS().run(df3.rdd)
val predictions = model.transform(df3)

Answer 1

您必须将df3.rdd [Row]映射到RDD [LabeledPoint]

所以这取决于你的架构和所有。