自动训练Embedding词向量和手动训练Embedding词向量

手动训练自己的词向量

gensim工具包-训练自己的Word2Vec
产生word_vector.bin文件
然后调用:

embedding = nn.Embedding.from_pretrained(word_vector.bin)

自动训练词向量

self.embed = nn.Embedding(vocab_size,embedding_dim)
#词向量需要跟随模型一起训练

你可能感兴趣的:(人工智能,Python,人工智能,机器学习,深度学习,nlp)