minor fix for Word2Vec test

xiliu82 · Aug 4, 2014 · 2ba9483 · 2ba9483
1 parent e248441
commit 2ba9483
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mllib/src/test/scala/org/apache/spark/mllib/feature/Word2VecSuite.scala b/mllib/src/test/scala/org/apache/spark/mllib/feature/Word2VecSuite.scala
@@ -36,7 +36,7 @@ class Word2VecSuite extends FunSuite with LocalSparkContext {
     val minCount = 2
     val num = 2
 
-    val model = Word2Vec.train(doc, size, startingAlpha, window, minCount)
+    val model = Word2Vec.train(doc, size, startingAlpha)
     val syms = model.findSynonyms("a", 2)
     assert(syms.length == num)
     assert(syms(0)._1 == "b")