gensim工具包-训练自己的Word2Vec
产生word_vector.bin文件
然后调用:
embedding = nn.Embedding.from_pretrained(word_vector.bin)自动训练词向量
self.embed = nn.Embedding(vocab_size,embedding_dim) #词向量需要跟随模型一起训练

gensim工具包-训练自己的Word2Vec
产生word_vector.bin文件
然后调用:
embedding = nn.Embedding.from_pretrained(word_vector.bin)自动训练词向量
self.embed = nn.Embedding(vocab_size,embedding_dim) #词向量需要跟随模型一起训练