[NLP] snunlp

nlp 환경에 접속

conda activate nlp

 

라이브러리 설치 

# 필수 라이브러리 설치
pip install torch transformers sentence-transformers

 

 

사용 가능한 모델 ID: snunlp/KR-SBERT-V40K-klueNLI-augSTS

 

이 모델은 “KLUE-NLI” 데이터에 기반해 파인튜닝된 SBERT로, 문장 유사도나 클러스터링 등에 바로 활용

 

from sentence_transformers import SentenceTransformer, util

# 올바른 모델 ID
model = SentenceTransformer("snunlp/KR-SBERT-V40K-klueNLI-augSTS")
sentences = ["문장 A", "문장 B"]

# 임베딩 생성 및 유사도 계산
embeddings = model.encode(sentences, convert_to_tensor=True)
cos_sim = util.pytorch_cos_sim(embeddings[0], embeddings[1])
print(f"Cosine similarity: {cos_sim.item():.4f}")

 

 

결과 

"오뚜기참기름", "오뜌기참기름"

Cosine similarity: 0.1739