-
- Downloads
[SPARK-13600][MLLIB] Use approxQuantile from DataFrame stats in QuantileDiscretizer
## What changes were proposed in this pull request? QuantileDiscretizer can return an unexpected number of buckets in certain cases. This PR proposes to fix this issue and also refactor QuantileDiscretizer to use approxQuantiles from DataFrame stats functions. ## How was this patch tested? QuantileDiscretizerSuite unit tests (some existing tests will change or even be removed in this PR) Author: Oliver Pierson <ocp@gatech.edu> Closes #11553 from oliverpierson/SPARK-13600.
Showing
- mllib/src/main/scala/org/apache/spark/ml/feature/QuantileDiscretizer.scala 25 additions, 94 deletions...ala/org/apache/spark/ml/feature/QuantileDiscretizer.scala
- mllib/src/test/scala/org/apache/spark/ml/feature/QuantileDiscretizerSuite.scala 40 additions, 75 deletions...rg/apache/spark/ml/feature/QuantileDiscretizerSuite.scala
Loading
Please register or sign in to comment