-
- Downloads
[SPARK-14147][ML][SPARKR] SparkR predict should not output feature column
## What changes were proposed in this pull request? SparkR does not support type of vector which is the default type of feature column in ML. R predict also does not output intermediate feature column. So SparkR ```predict``` should not output feature column. In this PR, I only fix this issue for ```naiveBayes``` and ```survreg```. ```kmeans``` has the right code route already and ```glm``` will be fixed at SparkRWrapper refactor(#12294). ## How was this patch tested? No new tests. cc mengxr shivaram Author: Yanbo Liang <ybliang8@gmail.com> Closes #11958 from yanboliang/spark-14147.
Showing
- mllib/src/main/scala/org/apache/spark/ml/r/AFTSurvivalRegressionWrapper.scala 1 addition, 1 deletion.../org/apache/spark/ml/r/AFTSurvivalRegressionWrapper.scala
- mllib/src/main/scala/org/apache/spark/ml/r/NaiveBayesWrapper.scala 3 additions, 1 deletion.../main/scala/org/apache/spark/ml/r/NaiveBayesWrapper.scala
Please register or sign in to comment