Kevin Hu
commited on
Commit
·
e6fc1a9
1
Parent(s):
e6abe77
refine using rerank model (#2553)
Browse files### What problem does this PR solve?
#2552
### Type of change
- [x] Performance Improvement
- rag/nlp/search.py +1 -1
rag/nlp/search.py
CHANGED
|
@@ -348,7 +348,7 @@ class Dealer:
|
|
| 348 |
ins_tw.append(tks)
|
| 349 |
|
| 350 |
tksim = self.qryr.token_similarity(keywords, ins_tw)
|
| 351 |
-
vtsim,_ = rerank_mdl.similarity(
|
| 352 |
|
| 353 |
return tkweight*np.array(tksim) + vtweight*vtsim, tksim, vtsim
|
| 354 |
|
|
|
|
| 348 |
ins_tw.append(tks)
|
| 349 |
|
| 350 |
tksim = self.qryr.token_similarity(keywords, ins_tw)
|
| 351 |
+
vtsim,_ = rerank_mdl.similarity(query, [rmSpace(" ".join(tks)) for tks in ins_tw])
|
| 352 |
|
| 353 |
return tkweight*np.array(tksim) + vtweight*vtsim, tksim, vtsim
|
| 354 |
|