Sentence transformers

Pywiki
둘러보기로 가기 검색하러 가기

1 기초 사용[편집 | 원본 편집]

의도 설명 방법
설치 pip install sentence_transformers
임포트 코드 내에서 다음과 같이 불러온다. from sentence_transformers import SentenceTransformer

2 한국어 임베딩[편집 | 원본 편집]

의도 설명 방법
임베딩 모델 생성 model = SentenceTransformer('jhgan/ko-sroberta-multitask')
임베딩 input에 텍스트 자료를 받아 진행.

768차원의 벡터를 반환한다.

embedding = model.encode(input)