I use Spark MLlib to train a domain specific word2vec model and I need to use it in glove word2vec format. How can I convert it to glove txt format?
Asked
Active
Viewed 110 times
1 Answers
0
After trying a bit in SparkShell, I found below code works for me:
val vectors = model.getVectors
val writer = new BufferedWriter(new FileWriter(file))
vectors foreach ( entry => writer.write(entry._1 + " " + entry._2.mkString(" ") + "\n") )

user3053227
- 63
- 2
- 8