티스토리 뷰
Doc2vec을 사용할때 문서의 이름(태그)를 같이 학습합니다. 이때 학습 후 모델을 사용할때 학습할떄 포함되지 않은 태그를 불러오게 되면 오류가 나옵니다.
저는 해당 모델에 태그의 포함유무를 확인하는 과정이 필요해서 학습된 doc2vec 모델에 포함된 태그 리스트를 가져와야 했습니다.
찾아보니 gensim에서 자체적으로 포함하고 있는 기능이어서 간단하게 정리해 둡니다.
tag_list = self.user_model.docvecs.index2entity
'기계학습' 카테고리의 다른 글
| [Gensim] doc2vec, word2vec 학습시 c complier 사용 여부 확인하기 (0) | 2020.05.02 |
|---|---|
| [Pytorch] nn.Embedding.from_pretrained로 fasttext 불러오기 (0) | 2020.04.19 |
| [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 |
댓글