0

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?

user3053227
  • 63
  • 2
  • 8

1 Answers1

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