-
- Downloads
[SPARK-18282][ML][PYSPARK] Add python clustering summaries for GMM and BKM
## What changes were proposed in this pull request? Add model summary APIs for `GaussianMixtureModel` and `BisectingKMeansModel` in pyspark. ## How was this patch tested? Unit tests. Author: sethah <seth.hendrickson16@gmail.com> Closes #15777 from sethah/pyspark_cluster_summaries. (cherry picked from commit e811fbf9) Signed-off-by:Yanbo Liang <ybliang8@gmail.com>
Showing
- mllib/src/main/scala/org/apache/spark/ml/classification/LogisticRegression.scala 5 additions, 6 deletions...g/apache/spark/ml/classification/LogisticRegression.scala
- mllib/src/main/scala/org/apache/spark/ml/clustering/BisectingKMeans.scala 4 additions, 5 deletions...cala/org/apache/spark/ml/clustering/BisectingKMeans.scala
- mllib/src/main/scala/org/apache/spark/ml/clustering/GaussianMixture.scala 4 additions, 5 deletions...cala/org/apache/spark/ml/clustering/GaussianMixture.scala
- mllib/src/main/scala/org/apache/spark/ml/clustering/KMeans.scala 4 additions, 5 deletions...rc/main/scala/org/apache/spark/ml/clustering/KMeans.scala
- mllib/src/main/scala/org/apache/spark/ml/regression/GeneralizedLinearRegression.scala 5 additions, 6 deletions...che/spark/ml/regression/GeneralizedLinearRegression.scala
- mllib/src/main/scala/org/apache/spark/ml/regression/LinearRegression.scala 7 additions, 7 deletions...ala/org/apache/spark/ml/regression/LinearRegression.scala
- mllib/src/test/scala/org/apache/spark/ml/classification/LogisticRegressionSuite.scala 2 additions, 0 deletions...che/spark/ml/classification/LogisticRegressionSuite.scala
- mllib/src/test/scala/org/apache/spark/ml/clustering/BisectingKMeansSuite.scala 3 additions, 0 deletions...org/apache/spark/ml/clustering/BisectingKMeansSuite.scala
- mllib/src/test/scala/org/apache/spark/ml/clustering/GaussianMixtureSuite.scala 3 additions, 0 deletions...org/apache/spark/ml/clustering/GaussianMixtureSuite.scala
- mllib/src/test/scala/org/apache/spark/ml/clustering/KMeansSuite.scala 3 additions, 0 deletions...st/scala/org/apache/spark/ml/clustering/KMeansSuite.scala
- mllib/src/test/scala/org/apache/spark/ml/regression/GeneralizedLinearRegressionSuite.scala 2 additions, 0 deletions...park/ml/regression/GeneralizedLinearRegressionSuite.scala
- mllib/src/test/scala/org/apache/spark/ml/regression/LinearRegressionSuite.scala 2 additions, 0 deletions...rg/apache/spark/ml/regression/LinearRegressionSuite.scala
- python/pyspark/ml/classification.py 9 additions, 6 deletionspython/pyspark/ml/classification.py
- python/pyspark/ml/clustering.py 159 additions, 3 deletionspython/pyspark/ml/clustering.py
- python/pyspark/ml/regression.py 12 additions, 4 deletionspython/pyspark/ml/regression.py
- python/pyspark/ml/tests.py 32 additions, 0 deletionspython/pyspark/ml/tests.py
Loading
Please register or sign in to comment