[Gensim] Pretrained Doc2vec model에서 tag 리스트 가져오기

Doc2vec을 사용할때 문서의 이름(태그)를 같이 학습합니다. 이때 학습 후 모델을 사용할때 학습할떄 포함되지 않은 태그를 불러오게 되면 오류가 나옵니다.

 

저는 해당 모델에 태그의 포함유무를 확인하는 과정이 필요해서 학습된 doc2vec 모델에 포함된 태그 리스트를 가져와야 했습니다.

찾아보니 gensim에서 자체적으로 포함하고 있는 기능이어서 간단하게 정리해 둡니다.

 

tag_list = self.user_model.docvecs.index2entity