from sentence_transformers import SparseEncoder
from sentence_transformers.sparse_encoder.evaluation import (
    SparseInformationRetrievalEvaluator,
)

model = SparseEncoder("rasyosef/splade-tiny", device="cpu")

queries = {
    "q1": "reset a forgotten password",
    "q2": "export invoices as CSV",
}

corpus = {
    "d1": "Reset a forgotten password from the account security page.",
    "d2": "Export invoices as a CSV file from the billing dashboard.",
    "d3": "Change the workspace color theme.",
    "d4": "Archive an inactive user account.",
}

relevant_docs = {
    "q1": {"d1"},
    "q2": {"d2"},
}