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 |
|