vrashad commited on
Commit
ffb41cf
·
verified ·
1 Parent(s): 62814dc

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -6
README.md CHANGED
@@ -62,20 +62,20 @@ from sentence_transformers import SentenceTransformer
62
 
63
  model = SentenceTransformer("LocalDoc/LocRet-small")
64
 
65
- queries = ["query: Azərbaycanın paytaxtı hansı şəhərdir?"]
66
  passages = [
67
- "passage: Bakı Azərbaycan Respublikasının paytaxtı və ən böyük şəhəridir.",
68
- "passage: Gəncə Azərbaycanın ikinci böyük şəhəridir.",
69
  ]
70
 
71
- query_embeddings = model.encode(queries)
72
- passage_embeddings = model.encode(passages)
73
 
74
  similarities = model.similarity(query_embeddings, passage_embeddings)
75
  print(similarities)
76
  ```
 
77
 
78
- > **Important:** Always use `"query: "` prefix for queries and `"passage: "` prefix for documents.
79
 
80
  ## Training
81
 
 
62
 
63
  model = SentenceTransformer("LocalDoc/LocRet-small")
64
 
65
+ queries = ["Azərbaycanın paytaxtı hansı şəhərdir?"]
66
  passages = [
67
+ "Bakı Azərbaycan Respublikasının paytaxtı və ən böyük şəhəridir.",
68
+ "Gəncə Azərbaycanın ikinci böyük şəhəridir.",
69
  ]
70
 
71
+ query_embeddings = model.encode_query(queries)
72
+ passage_embeddings = model.encode_document(passages)
73
 
74
  similarities = model.similarity(query_embeddings, passage_embeddings)
75
  print(similarities)
76
  ```
77
+ Prefixes `"query: "` and `"passage: "` are applied automatically via encode_query and encode_document. If using model.encode directly, the `"passage: "` prefix is added by default.
78
 
 
79
 
80
  ## Training
81