Alexander Ulanov created SPARK-9120:
---------------------------------------
Summary: Add multivariate regression (or prediction) interface
Key: SPARK-9120
URL: https://issues.apache.org/jira/browse/SPARK-9120
Project: Spark
Issue Type: Improvement
Components: ML
Affects Versions: 1.4.0
Reporter: Alexander Ulanov
Fix For: 1.4.0
org.apache.spark.ml.regression.RegressionModel supports prediction only for a
single variable with a method "predict:Double" by extending the Predictor.
There is a need for multivariate prediction, at least for regression. I propose
to modify "RegressionModel" interface similarly to how it is done in
"ClassificationModel", which supports multiclass classification. It has
"predict:Double" and "predictRaw:Vector". Analogously, "RegressionModel" should
have something like "predictMultivariate:Vector".
--
This message was sent by Atlassian JIRA
(v6.3.4#6332)
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
