Спарк (1,6) ML Linear Regression - как предсказать ж/модель

Question

У меня есть рабочая модель линейной регрессии:

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

и у меня есть данные в dataframe:

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, чтобы иметь одинаковую структуру ваших данных обучения, следовательно, столбец .