sentence-transformers How to use strongpear/M3-retriever-Wiki with sentence-transformers:
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("strongpear/M3-retriever-Wiki")
sentences = [
"Thành phố San Fernando là một phần của nước nào và có bao nhiêu dân số?",
"Boule-d'Amont là một xã trong vùng Occitanie, thuộc tỉnh Pyrénées-Orientales, quận Prades, tổng Vinça. Tọa độ địa lý của xã là 42° 34' vĩ độ bắc, 02° 36' kinh độ đông. Boule-d'Amont nằm trên độ cao trung bình là 483 mét trên mực nước biển, có điểm thấp nhất là 234 mét và điểm cao nhất là 1.348 mét. Xã có diện tích 23,22 km², dân số vào thời điểm 1999 là 73 người; mật độ dân số là 3 người/km².",
"Thành phố San Fernando là một trong ba thành phố lớn của Trinidad và Tobago, là đô thị lớn thứ hai. Thành phố có diện tịch 18 km² và nằm ở phía tây nam của đảo Trinidad. Nó giáp sông Guaracara về phía bắc, giáp sông Oropouche về phía nam, giáp quốc lộ Sir Solomon Hochoy về phía đông, và vịnh Paria về phía tây. Dân số ước tính năm 2000 là 62.000 người. Đô thị San Fernando được nâng cấp thành thành phố ngày 18 tháng 11 năm 1988. Khẩu hiệu của San Fernando là: \"Sanitas Fortis\" - \"Trong một môi trường lành mạnh chúng ta sẽ tìm thấy sức mạnh\".",
"Các dự án quân sự bản địa khác liên quan đến việc thiết kế và bổ sung hàng không mẫu hạm lớp \"Vikrant\" và tàu ngầm hạt nhân lớp \"Arihant\".\nSau khi kết thúc Chiến tranh Lạnh,Ấn Độ tăng cường hợp tác về kinh tế,chiến lược và quân sự với Hoa Kỳ và Liên minh châu Âu. Năm 2008, Hoa Kỳ và Ấn Độ ký kết một thỏa thuận hạt nhân dân sự.Mặc dù đương thời Ấn Độ là quốc gia sở hữu vũ khí hạt nhân và không phải là một bên tham gia của Hiệp ước không phổ biến vũ khí hạt nhân,song quốc gia này vẫn nhận được sự miễn trừ từ Cơ quan Năng lượng Nguyên tử Quốc tế và Nhóm các nhà cung cấp hạt nhân,do vậy thoát khỏi các hạn chế,rào cản trước đây đối với công nghệ và thương mại hạt nhân. Như một hệ quả, Ấn Độ trở thành quốc gia thứ sáu sở hữu vũ khí hạt nhân \"trên thực tế\". Ấn Độ sau đó ký kết các thỏa thuận hợp tác liên quan đến năng lượng hạt nhân dân sự với Nga, Pháp, Anh Quốc, và Canada.\nTổng thống Ấn Độ là thống soái tối cao của lực lượng vũ trang quốc gia với 1,6 triệu quân tại ngũ và xếp thứ ba thế giới trên tiêu chí này. Quân đội Ấn Độ gồm có lục quân, hải quân, và không quân; các tổ chức phụ trợ gồm có Bộ tư lệnh chiến lược (Strategic Forces Command) và ba nhóm bán quân sự: Đội quân súng trường Assam, Lực lượng biên cảnh đặc chủng, và Lực lượng bảo vệ bờ biển Ấn Độ. Ngân sách quốc phòng chính thức của Ấn Độ giai đoạn 2012-17 chiếm khoảng 2,5% GDP. Năm 2012, Ấn Độ là nước nhập khẩu vũ khí lớn nhất thế giới;từ năm 2007 đến năm 2011, tiền mua vũ khí của Ấn Độ chiếm 10% tổng phí tổn dành cho mua sắm vũ khí. Phần lớn chi tiêu quân sự tập trung vào phòng thủ đối với riêng Pakistan và chống lại ảnh hưởng ngày càng gia tăng của Trung Quốc tại Ấn Độ Dương.\nMột trận ẩu đả dữ dội giữa quân đội Trung Quốc và Ấn Độ tại khu vực lãnh thổ tranh chấp đã làm 20 binh sĩ Ấn Độ và ít nhất 4 binh sĩ Trung Quốc thiệt mạng."
]
embeddings = model.encode(sentences)
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [4, 4] 
