我可以创建一个带有预训练嵌入的Apache Spark Word2VecModel吗?

时间:2017-12-03 21:59:37

标签: apache-spark pyspark word2vec apache-spark-ml word-embedding

我想在我的Python Spark文本分类管道中使用单词嵌入。 documentation显示了如何训练自己的嵌入,但我想使用像GloVe这样的预训练集。有没有办法用预训练嵌入来初始化Word2VecModel,还是我必须编写自己的变换器类?

0 个答案:

没有答案