python 使用word2vec词嵌入代码
时间: 2024-01-22 21:02:56 浏览: 160
课程设计 基于Python+Word2Vec的近义词分析系统源码+全部资料齐全.zip
5星 · 资源好评率100%
以下是使用Python中gensim库实现Word2Vec词嵌入的代码:
```python
from gensim.models import Word2Vec
# 定义语料库
sentences = [['this', 'is', 'the', 'first', 'sentence', 'for', 'word2vec'],
['this', 'is', 'the', 'second', 'sentence'],
['yet', 'another', 'sentence'],
['one', 'more', 'sentence'],
['and', 'the', 'final', 'sentence']]
# 训练模型
model = Word2Vec(sentences, min_count=1)
# 获取词向量
vector = model.wv['sentence']
print(vector)
```
上述代码中,我们首先定义了一个语料库,然后使用该语料库训练了一个Word2Vec模型。接着,我们可以使用`model.wv`获取词向量,例如获取单词“sentence”的向量表示。最后,我们将该向量打印出来。
阅读全文