-
- Downloads
[SPARK-18625][ML] OneVsRestModel should support setFeaturesCol and setPredictionCol
## What changes were proposed in this pull request? add `setFeaturesCol` and `setPredictionCol` for `OneVsRestModel` ## How was this patch tested? added tests Author: Zheng RuiFeng <ruifengz@foxmail.com> Closes #16059 from zhengruifeng/ovrm_setCol. (cherry picked from commit bdfe7f67) Signed-off-by:Yanbo Liang <ybliang8@gmail.com>
Showing
- mllib/src/main/scala/org/apache/spark/ml/classification/OneVsRest.scala 9 additions, 0 deletions.../scala/org/apache/spark/ml/classification/OneVsRest.scala
- mllib/src/test/scala/org/apache/spark/ml/classification/OneVsRestSuite.scala 13 additions, 1 deletion...a/org/apache/spark/ml/classification/OneVsRestSuite.scala
Loading
Please register or sign in to comment