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