Use a more accurate model for symmetric semantic search

- 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

View File

@@ -79,7 +79,7 @@ default_config = {
{
'symmetric':
{
'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': None
},