火花（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列。