splade4elastic

Splade4elastic

Pembungkus ekspansi kueri sederhana untuk pencarian elastis, yang menggunakan bobot kustom kata kunci yang berasal dari transformator model berbahasa bertopeng permukaan pelukan.

Contoh penggunaan

 from splade4elastic import SpladeRewriter
model_name = "roberta-base"
splader = SpladeRewriter(model_name)
test_texts = [
    "My name is John",
    "The quick brown fox jumps over the lazy dog",
    "I like to eat apples",
]
for test_text in test_texts:
    print(test_text)
    print(splader.query_expand(test_text))