sentence-transformers How to use meandyou200175/pB-query-sql with sentence-transformers:
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("meandyou200175/pB-query-sql")
sentences = [
"mình cần máy giặt dung tích giặt lớn hơn 4kg",
"Máy giặt LG Inverter 4kg, Công nghệ AI DD, Giặt hơi nước, Giá: 7.900.000",
"Đồng hồ Apple Watch Series 9, Pin 18 giờ, Màn OLED 1.9\", Giá: 11.500.000",
"Máy giặt LG Inverter 2kg, Công nghệ AI DD, Giặt hơi nước, Giá: 7.900.000",
"Tủ lạnh Toshiba Inverter 180L, Ngăn đá trên, Giá: 5.400.000",
"Laptop LG Gram 14, Nặng 1.1kg, RAM 16GB, SSD 512GB, Giá: 19.800.000",
"Laptop Dell XPS 15, SSD 1TB, RAM 16GB, Core i7, Giá: 42.000.000",
"Tủ rượu Kadeka KS78TL, Sức chứa 72 chai, Cửa kính chống UV, Giá: 19.500.000",
"Quạt bàn Mitsubishi D16-GV CY, công suất 45W, 3 tốc độ gió, tiết kiệm điện, Giá 620000",
"Máy giặt LG Inverter 9kg, Công nghệ AI DD, Giặt hơi nước, Giá: 7.900.000",
"Xe tay ga Honda Air Blade 160, Động cơ 156cc, Phun xăng điện tử, Giá: 55.000.000",
"Máy ảnh Canon EOS M50 Mark II, 24.1MP, Nặng 580g, Lens kit 15-45mm, Giá: 17.800.000",
"Bình nước nóng Ariston SL30, Dung tích 30L, Thanh đốt Inox, Giá: 2.800.000",
"Tủ giày gỗ công nghiệp, Ngang 85cm, 3 tầng, Giá: 1.500.000",
"Máy giặt LG Inverter 3kg, Công nghệ AI DD, Giặt hơi nước, Giá: 7.900.000",
"Máy xay Philips HR2115, công suất 515W, cối 1.5L, Giá: 1.203.000",
"Robot hút bụi Roborock Q7 Max, Độ ồn 50dB, Pin 5200mAh, Giá: 10.200.000"
]
embeddings = model.encode(sentences)
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [17, 17]