火花（1.6）ML線性迴歸 - 如何預測瓦特/模型

Question

我有一個工作的線性迴歸模型：

lrModel 
org.apache.spark.ml.regression.LinearRegressionModel 

和我在一個數據幀有數據：

data 
org.apache.spark.sql.DataFrame = [label: double, features: vector] 

怎麼辦我使用模型來預測？在我的情況，我想要做的事，如：

lrModel.predict(data) // which doesn't work 

則預期值（標籤）比較預測值

Answer 1

要預測你需要有一個Dataframe，並使用transform方法將其轉換這是所有ML Models的一部分。請注意，它們都需要DataFrame s具有與您的培訓數據相同的結構，因此需要fetures列。