Open
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
Metadata
Metadata
Assignees
Labels
No labels