PyTorch / Gensim - How to load pre-trained word embeddings

后端 未结 6 775
甜味超标
甜味超标 2020-12-01 02:44

I want to load a pre-trained word2vec embedding with gensim into a PyTorch embedding layer.

So my question is, how do I get the embedding weights loaded by gensim in

6条回答
  •  半阙折子戏
    2020-12-01 03:32

    from gensim.models import Word2Vec
    
    model = Word2Vec(reviews,size=100, window=5, min_count=5, workers=4)
    #gensim model created
    
    import torch
    
    weights = torch.FloatTensor(model.wv.vectors)
    embedding = nn.Embedding.from_pretrained(weights)
    

提交回复
热议问题