diff options
author | Travis Galoppo <tjg2107@columbia.edu> | 2015-01-20 12:58:11 -0800 |
---|---|---|
committer | Xiangrui Meng <meng@databricks.com> | 2015-01-20 12:58:11 -0800 |
commit | 23e25543beaa5966b5f07365f338ce338fd6d71f (patch) | |
tree | 8c0642e5e895a02b0d97f1a2f5ca45142751fab1 /examples/src/main | |
parent | 769aced9e7f058f5008ce405f7c9714c3db203be (diff) | |
download | spark-23e25543beaa5966b5f07365f338ce338fd6d71f.tar.gz spark-23e25543beaa5966b5f07365f338ce338fd6d71f.tar.bz2 spark-23e25543beaa5966b5f07365f338ce338fd6d71f.zip |
SPARK-5019 [MLlib] - GaussianMixtureModel exposes instances of MultivariateGauss...
This PR modifies GaussianMixtureModel to expose instances of MutlivariateGaussian rather than separate mean and covariance arrays.
Author: Travis Galoppo <tjg2107@columbia.edu>
Closes #4088 from tgaloppo/spark-5019 and squashes the following commits:
3ef6c7f [Travis Galoppo] In GaussianMixtureModel: Changed name of weight, gaussian to weights, gaussians. Other sources modified accordingly.
091e8da [Travis Galoppo] SPARK-5019 - GaussianMixtureModel exposes instances of MultivariateGaussian rather than mean/covariance matrices
Diffstat (limited to 'examples/src/main')
-rw-r--r-- | examples/src/main/scala/org/apache/spark/examples/mllib/DenseGmmEM.scala | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/examples/src/main/scala/org/apache/spark/examples/mllib/DenseGmmEM.scala b/examples/src/main/scala/org/apache/spark/examples/mllib/DenseGmmEM.scala index 948c350953..de58be38c7 100644 --- a/examples/src/main/scala/org/apache/spark/examples/mllib/DenseGmmEM.scala +++ b/examples/src/main/scala/org/apache/spark/examples/mllib/DenseGmmEM.scala @@ -54,7 +54,7 @@ object DenseGmmEM { for (i <- 0 until clusters.k) { println("weight=%f\nmu=%s\nsigma=\n%s\n" format - (clusters.weight(i), clusters.mu(i), clusters.sigma(i))) + (clusters.weights(i), clusters.gaussians(i).mu, clusters.gaussians(i).sigma)) } println("Cluster labels (first <= 100):") |