将Spark MLlib Word2Vec模型转换为GloVe txt格式的步骤如下:
from pyspark.ml.feature import Word2VecModel
# 加载Word2Vec模型
word2vec_model = Word2VecModel.load("path/to/word2vec_model")
# 导出模型为文本格式
word2vec_model.getVectors().write.text("path/to/output_dir")
make
命令编译GloVe工具。./glove_converter.sh input_dir output_dir
其中,input_dir
是导出的Word2Vec模型的文本格式所在的目录,output_dir
是转换后的GloVe txt文件的输出目录。
没有搜到相关的沙龙
领取专属 10元无门槛券
手把手带您无忧上云