유사도 검색

유사도 검색

문서를 벡터로 바꿨다면, 이제 질문(쿼리)과 비슷한 문서를 찾는 단계예요. Gensim은 similarities 모듈로 벡터 공간 안에서 관련 문서를 효율적으로 검색할 수 있어요.

기본 흐름

from gensim.similarities import MatrixSimilarity

index = MatrixSimilarity(model[corpus])
sims = index[model[query_doc]]  # 쿼리와의 유사도 벡터

확인 포인트

  • 문서를 모델로 변환한 뒤 인덱스에 넣고, 쿼리도 같은 변환으로 만들어 비교해요.
  • 코퍼스가 크면 인덱스 생성이 첫 단계에서 한 번만 일어나요.

출처: https://radimrehurek.com/gensim/auto_tutorials/core/4_similarity_queries.html

더 알아보기