기계학습
[Gensim] doc2vec, word2vec 학습시 c complier 사용 여부 확인하기
Demis Tae Kyu Eom
2020. 5. 2. 00:23
gensim 패키지를 사용해서 doc2vec 모델이나 word2vec 모델을 학습할 때에 gensim 자체적으로 c complier를 사용하여 더욱 빠르게 학습이 가능하게 하였다고 합니다.
하지만 자신의 컴퓨터에 c extension이 설치 되지 않은 경우에는 학습 속도가 느릴것이다는 메시지가 나올수 있습니다. 해당 메시지를 보지 않고 미리 C complier와 link가 잘 되었는지 확인하는 방법을 간단하게 정리합니다.
from gensim.models.word2vec import FAST_VERSION
위와 같이 FAST_VERSION을 import 한 후에 해당 값이 '0' 혹은 '1'인 경우에는 C complier를 사용해서 빠른 속도로 학습할 준비가 된것을 확인 할 수 있습니다.
데이터가 작은 경우네는 큰 차이가 없겠지만 속도면에서 최대 70배 차이가 난다고하니 가능하면 설정을 하고 학습하는게 좋을거 같습니다.