splade4elastic

Stade4élastique

Un simple emballage d'extension de requête pour la recherche élastique, qui utilise des poids personnalisés de mots clés dérivés d'un transformateur de modèle de langage masqué.

Exemple d'utilisation

 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))