pyspark - 如何在 payspark ml 中加载 word2vec 模型？

Question

我尝试使用以下几行加载经过训练的 word2vec 模型

from pyspark.ml.feature import Word2Vec

loadedWord2Vec = Word2Vec.load(W2V_MODEL_PATH)

但我收到此错误：

java.lang.NoSuchMethodException: org.apache.spark.ml.feature.Word2VecModel.<init>(java.lang.String)

标签： pysparkword2vec

Answer 1

我使用 Word2VecModel 而不是 Word2Vec 来加载模型并解决问题。