티스토리 뷰
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))
'기계학습' 카테고리의 다른 글
[Gensim] Pretrained Doc2vec model에서 tag 리스트 가져오기 (0) | 2020.05.07 |
---|---|
[Gensim] doc2vec, word2vec 학습시 c complier 사용 여부 확인하기 (0) | 2020.05.02 |
[Gensim] unable to import 'smart_open.gcs', disabling that module 오류 해결 (0) | 2020.04.19 |
[Pytorch]“AttributeError: LSTM object has no attribute flat_weights_names” (0) | 2020.03.31 |
[Pytorch] F.cross entropy vs nn.Cross Entropy loss (0) | 2020.02.27 |
댓글