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

Files changed (1) hide show
  1. 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(" ".join(keywords), [rmSpace(" ".join(tks)) for tks in ins_tw])
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