【发布时间】:2018-12-21 23:36:29
【问题描述】:
我使用 Spark MLlib 来训练特定领域的 word2vec 模型,我需要以 glove word2vec 格式使用它。 如何将其转换为手套 txt 格式?
【问题讨论】:
标签: apache-spark apache-spark-mllib word-embedding glove
我使用 Spark MLlib 来训练特定领域的 word2vec 模型,我需要以 glove word2vec 格式使用它。 如何将其转换为手套 txt 格式?
【问题讨论】:
标签: apache-spark apache-spark-mllib word-embedding glove
在 SparkShell 中尝试了一下之后,我发现下面的代码对我有用:
val vectors = model.getVectors
val writer = new BufferedWriter(new FileWriter(file))
vectors foreach ( entry => writer.write(entry._1 + " " + entry._2.mkString(" ") + "\n") )
【讨论】: