We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 26a948d commit e248441Copy full SHA for e248441
mllib/src/test/scala/org/apache/spark/mllib/feature/Word2VecSuite.scala
@@ -47,10 +47,10 @@ class Word2VecSuite extends FunSuite with LocalSparkContext {
47
test("Word2VecModel") {
48
val num = 2
49
val localModel = Seq(
50
- ("china" , Array(0.50f, 0.50f, 0.50f, 0.50f)),
51
- ("japan" , Array(0.40f, 0.50f, 0.50f, 0.50f)),
52
- ("taiwan", Array(0.60f, 0.50f, 0.50f, 0.50f)),
53
- ("korea" , Array(0.45f, 0.60f, 0.60f, 0.60f))
+ ("china", Array(0.50f, 0.50f, 0.50f, 0.50f)),
+ ("japan", Array(0.40f, 0.50f, 0.50f, 0.50f)),
+ ("taiwan", Array(0.60f, 0.50f, 0.50f, 0.50f)),
+ ("korea", Array(0.45f, 0.60f, 0.60f, 0.60f))
54
)
55
val model = new Word2VecModel(sc.parallelize(localModel, 2))
56
val syms = model.findSynonyms("china", num)
0 commit comments