我想word2vec模型训练的所有word,该怎么写
时间: 2024-02-17 14:01:50 浏览: 61
百度百科新闻小说word2vec预训练中文模型
如果你使用的是 Gensim 库来训练 Word2Vec 模型,那么可以通过 `model.wv.index_to_key` 属性获取模型中所有的词汇,返回的是一个列表,包含模型中所有词汇的字符串表示,你可以按照如下方式使用:
```python
from gensim.models import Word2Vec
# 假设你已经训练好了一个 Word2Vec 模型
model = Word2Vec.load('your_model_path')
# 获取模型中所有的词汇
all_words = model.wv.index_to_key
# 打印所有词汇
for word in all_words:
print(word)
```
这样就可以获取训练好的 Word2Vec 模型中的所有词汇了。注意,如果你使用的是旧版本的 Gensim 库,可能需要使用 `model.index2word` 属性来获取所有词汇。
阅读全文