splade4elastic

Fallido 4elástico

Un envoltorio de expansión de consulta simple para la búsqueda elástica, que utiliza pesos personalizados de palabras clave derivadas de un transformador de modelo enmascarado enmascarado de la cara de abrazos.

Uso de ejemplo

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