-
- Downloads
[SPARK-9493] [ML] add featureIndex to handle vector features in IsotonicRegression
This PR contains the following changes: * add `featureIndex` to handle vector features (in order to chain isotonic regression easily with output from logistic regression * make getter/setter names consistent with params * remove inheritance from Regressor because it is tricky to handle both `DoubleType` and `VectorType` * simplify test data generation jkbradley zapletal-martin Author: Xiangrui Meng <meng@databricks.com> Closes #7952 from mengxr/SPARK-9493 and squashes the following commits: 8818ac3 [Xiangrui Meng] address comments 05e2216 [Xiangrui Meng] address comments 8d08090 [Xiangrui Meng] add featureIndex to handle vector features make getter/setter names consistent with params remove inheritance from Regressor
Showing
- mllib/src/main/scala/org/apache/spark/ml/regression/IsotonicRegression.scala 146 additions, 56 deletions...a/org/apache/spark/ml/regression/IsotonicRegression.scala
- mllib/src/test/scala/org/apache/spark/ml/regression/IsotonicRegressionSuite.scala 48 additions, 34 deletions.../apache/spark/ml/regression/IsotonicRegressionSuite.scala
Loading
Please register or sign in to comment