Переглянути джерело

refine using rerank model (#2553)

### What problem does this PR solve?

#2552

### Type of change

- [x] Performance Improvement
tags/v0.12.0
Kevin Hu 1 рік тому
джерело
коміт
ecf441c830
Аккаунт користувача з таким Email не знайдено
1 змінених файлів з 1 додано та 1 видалено
  1. 1
    1
      rag/nlp/search.py

+ 1
- 1
rag/nlp/search.py Переглянути файл

@@ -348,7 +348,7 @@ class Dealer:
ins_tw.append(tks)

tksim = self.qryr.token_similarity(keywords, ins_tw)
vtsim,_ = rerank_mdl.similarity(" ".join(keywords), [rmSpace(" ".join(tks)) for tks in ins_tw])
vtsim,_ = rerank_mdl.similarity(query, [rmSpace(" ".join(tks)) for tks in ins_tw])

return tkweight*np.array(tksim) + vtweight*vtsim, tksim, vtsim


Завантаження…
Відмінити
Зберегти