如何在Spark中使用LogisticRegressionWIthLBFGS和Dataframe

Question

我希望在設計使用ML算法和數據框的程序中使用MLLIB LogisticRegressionWithLBFGS。 有沒有一種簡單的方法來將我的數據框轉換爲RDD而無需再次編寫新程序？ 像這樣的東西不起作用：

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

Answer 1

你必須映射您df3.rdd [行]對RDD [LabeledPoint]

所以，這取決於你的架構和所有。