Skip to content

Commit 2ba9483

Browse files
author
Liquan Pei
committed
minor fix for Word2Vec test
1 parent e248441 commit 2ba9483

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

mllib/src/test/scala/org/apache/spark/mllib/feature/Word2VecSuite.scala

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@ class Word2VecSuite extends FunSuite with LocalSparkContext {
3636
val minCount = 2
3737
val num = 2
3838

39-
val model = Word2Vec.train(doc, size, startingAlpha, window, minCount)
39+
val model = Word2Vec.train(doc, size, startingAlpha)
4040
val syms = model.findSynonyms("a", 2)
4141
assert(syms.length == num)
4242
assert(syms(0)._1 == "b")

0 commit comments

Comments
 (0)