fix rerank
This commit is contained in:
parent
63131ec9b2
commit
fc976efec7
1 changed files with 1 additions and 1 deletions
|
|
@ -347,7 +347,7 @@ class Dealer:
|
||||||
## For rank feature(tag_fea) scores.
|
## For rank feature(tag_fea) scores.
|
||||||
rank_fea = self._rank_feature_scores(rank_feature, sres)
|
rank_fea = self._rank_feature_scores(rank_feature, sres)
|
||||||
|
|
||||||
return tkweight * (np.array(tksim)+rank_fea) + vtweight * vtsim, tksim, vtsim
|
return tkweight * np.array(tksim) + vtweight * vtsim + rank_fea, tksim, vtsim
|
||||||
|
|
||||||
def hybrid_similarity(self, ans_embd, ins_embd, ans, inst):
|
def hybrid_similarity(self, ans_embd, ins_embd, ans, inst):
|
||||||
return self.qryr.hybrid_similarity(ans_embd,
|
return self.qryr.hybrid_similarity(ans_embd,
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue