We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent d5b36ab commit fa0a142Copy full SHA for fa0a142
1 file changed
mllib/src/main/scala/org/apache/spark/mllib/api/python/PythonMLLibAPI.scala
@@ -306,6 +306,7 @@ class PythonMLLibAPI extends Serializable {
306
var wtArray:Array[Double] = Array()
307
var muArray:Array[Vector] = Array()
308
var siArray :Array[Matrix] = Array()
309
+
310
for (i <- 0 until model.k) {
311
wtArray = wtArray ++ Array(model.weights(i))
312
muArray = muArray ++ Array(model.gaussians(i).mu)
@@ -329,6 +330,7 @@ class PythonMLLibAPI extends Serializable {
329
330
val weight = wt.asInstanceOf[Array[Double]]
331
val mean = mu.map(_.asInstanceOf[DenseVector])
332
val sigma = si.map(_.asInstanceOf[DenseMatrix])
333
334
val gaussians = Array.tabulate(weight.length){
335
i => new MultivariateGaussian(mean(i),sigma(i))
336
}
0 commit comments