aboutsummaryrefslogtreecommitdiff
path: root/mllib/src/test/java/org/apache
diff options
context:
space:
mode:
Diffstat (limited to 'mllib/src/test/java/org/apache')
-rw-r--r--mllib/src/test/java/org/apache/spark/mllib/clustering/JavaLDASuite.java6
1 files changed, 5 insertions, 1 deletions
diff --git a/mllib/src/test/java/org/apache/spark/mllib/clustering/JavaLDASuite.java b/mllib/src/test/java/org/apache/spark/mllib/clustering/JavaLDASuite.java
index b48f190f59..d272a42c85 100644
--- a/mllib/src/test/java/org/apache/spark/mllib/clustering/JavaLDASuite.java
+++ b/mllib/src/test/java/org/apache/spark/mllib/clustering/JavaLDASuite.java
@@ -19,6 +19,7 @@ package org.apache.spark.mllib.clustering;
import java.io.Serializable;
import java.util.ArrayList;
+import java.util.Arrays;
import scala.Tuple2;
@@ -59,7 +60,10 @@ public class JavaLDASuite implements Serializable {
@Test
public void localLDAModel() {
- LocalLDAModel model = new LocalLDAModel(LDASuite$.MODULE$.tinyTopics());
+ Matrix topics = LDASuite$.MODULE$.tinyTopics();
+ double[] topicConcentration = new double[topics.numRows()];
+ Arrays.fill(topicConcentration, 1.0D / topics.numRows());
+ LocalLDAModel model = new LocalLDAModel(topics, Vectors.dense(topicConcentration), 1D, 100D);
// Check: basic parameters
assertEquals(model.k(), tinyK);