bge-m3-poi-retrieval-ko

pip install -U sentence-transformers

from sentence_transformers import SentenceTransformer

# Download from the 🤗 Hub
model = SentenceTransformer("sentence_transformers_model_id")
# Run inference
sentences = [
    '창원중앙동',
    '경남 창원시 중앙구 중앙동 중앙동 | 창원시 중앙동 | 중앙구 중앙동 | 중앙동 중앙동',
    '경남 양산시 평산동 삼룡횟집 | 양산시 삼룡횟집 | 평산동 삼룡횟집',
]
embeddings = model.encode(sentences)
print(embeddings.shape)
# [3, 1024]

# Get the similarity scores for the embeddings
similarities = model.similarity(embeddings, embeddings)
print(similarities)
# tensor([[1.0000, 0.8642, 0.0517],
#         [0.8642, 1.0000, 0.0176],
#         [0.0517, 0.0176, 1.0000]])