기계학습
[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))