klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-07 21:29:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use a more accurate model for symmetric semantic search

Browse Source 
- The all-MiniLM-L6-v2 is more accurate
  - The exact previous model isn't benchmarked but based on the
    performance of the closest model to it. Seems like the new model
    maybe similar in speed and size

- On very preliminary evaluation of the model, the new model seems
  faster, with pretty decent results
This commit is contained in:
Debanjum Singh Solanky
2022-07-18 20:16:40 +04:00
parent 4a90972e38
commit 989526ae54
4 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/conftest.py
View File

@@ -14,7 +14,7 @@ def search_config(tmp_path_factory):
search_config = SearchConfig()
search_config.asymmetric = SymmetricSearchConfig(
encoder = "sentence-transformers/paraphrase-MiniLM-L6-v2",
encoder = "sentence-transformers/all-MiniLM-L6-v2",
cross_encoder = "cross-encoder/ms-marco-MiniLM-L-6-v2",
model_directory = model_dir
)