AttributeError: ‘KeyedVectors‘ object has no attribute ‘wv‘

错误原因:

Gensim 3.8.0 到 Gensim 4.0.0的更新,去除了wv,更改了部分函数名

gensim的word2vec示例如下:

save和load模块不变:

import gensim
from gensim import models

# self.w2v的定义和训练过程省略
...
# self.w2v的保存./bin文件
self.w2v.wv.save_word2vec_format('./model/w2v.bin', binary=False)
# 使用gensim进行load
self.w2v = models.KeyedVectors.load_word2vec_format('./model/w2v.bin', binary=False)

修改部分内容:

(1)
w2v_model.wv.get_vector(x)
->
w2v_model.get_vector(x)

(2)
w2v_model.wv.vocab.keys()
->
w2v_model.key_to_index.keys()

至此,解决问题!

猜你喜欢

转载自blog.csdn.net/buluxianfeng/article/details/125335804
今日推荐