cherche icon indicating copy to clipboard operation
cherche copied to clipboard

k param when creating sbert retriever not taken into account

Open fpservant opened this issue 2 years ago • 1 comments

Create a retriever based on a sentence bert, passing a value, eg. 10, to k param. It is not taken into account when calling the retriever (more values are returned)

    retriever = retrieve.Encoder(
        key='id',
        on=['content'],
        encoder=SentenceTransformer('sentence-transformers/all-MiniLM-L12-v2').encode,
        k = 10
    )
    retriever(documents=docs)

len(retriever(queries)[0]) > 10

fpservant avatar Sep 07 '23 22:09 fpservant