Repository navigation
k param when creating sbert retriever not taken into account #18
Description
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
Activity
fpservant commentedon Sep 8, 2023
Hum, seems to be same kind of problem as fe08cc8