기계학습

[Pytorch] nn.Embedding.from_pretrained로 fasttext 불러오기

Demis Tae Kyu Eom 2020. 4. 19. 18:29
from torch.nn import Embedding
import gensim

model_test = gensim.models.fasttext.load_facebook_model("/MY/FASTTEXT/bin MODEL FILE")
fastext_wv = model_test.wv.vectors

embeding = torch.nn.Embedding.from_pretrained(torch.FloatTensor(fastext_wv))