как я могу загрузить модель word2ve c в payspark мл?

Question

Я пытаюсь загрузить обученную модель word2ve c следующими строками

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)

Answer 1

Я использовал Word2VecModel вместо Word2Ve c для загрузки модели и проблема решена.