Views
No views yet
1from sentence_transformers import CrossEncoder
2
3model = CrossEncoder('ctu-aic/CE-fernet-c5-MRank256', max_length=256)
4
5query = "example query"
6
7documents = [
8 "Example document one.",
9 "Example document two.",
10 "Example document three."
11]
12
13top_k = 3
14return_documents = True
15
16results = model.rank(
17 query=query,
18 documents=documents,
19 top_k=top_k,
20 return_documents=return_documents
21)
22
23for i, res in enumerate(results):
24 print(f"{i+1}. {res['text']}")