Skip to content

Commit

Permalink
minor style change
Browse files Browse the repository at this point in the history
  • Loading branch information
Liquan Pei committed Aug 4, 2014
1 parent 26a948d commit e248441
Showing 1 changed file with 4 additions and 4 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -47,10 +47,10 @@ class Word2VecSuite extends FunSuite with LocalSparkContext {
test("Word2VecModel") {
val num = 2
val localModel = Seq(
("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))
("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))
)
val model = new Word2VecModel(sc.parallelize(localModel, 2))
val syms = model.findSynonyms("china", num)
Expand Down

0 comments on commit e248441

Please sign in to comment.