snowykami/meilisearch
1
0
Fork 0
mirror of https://github.com/meilisearch/meilisearch.git synced 2024-11-30 09:04:59 +08:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

Fix attributes score

Browse Source
This commit is contained in:
many 2021-03-24 18:20:13 +01:00
parent 0ad9499b93
commit ab92c814c3
No known key found for this signature in database
GPG Key ID: 2CEF23B75189EACA
1 changed files with 13 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
milli/src/search/criteria/attribute.rs
View File

@ -122,7 +122,8 @@ fn linear_compute_candidates(
fn compute_candidate_rank(branches: &Vec<Vec<Query>>, words_positions: HashMap<String, RoaringBitmap>) -> u64 { fn compute_candidate_rank(branches: &Vec<Vec<Query>>, words_positions: HashMap<String, RoaringBitmap>) -> u64 {
let mut min_rank = u64::max_value(); let mut min_rank = u64::max_value();
for branch in branches { for branch in branches {
let mut branch_rank = 0; let branch_len = branch.len();
let mut branch_rank = Vec::with_capacity(branch_len);
for Query { prefix, kind } in branch { for Query { prefix, kind } in branch {
// find the best position of the current word in the document. // find the best position of the current word in the document.
let position = match kind { let position = match kind {
@ -145,13 +146,21 @@ fn linear_compute_candidates(
// if a position is found, we add it to the branch score, // if a position is found, we add it to the branch score,
// otherwise the branch is considered as unfindable in this document and we break. // otherwise the branch is considered as unfindable in this document and we break.
if let Some(position) = position { if let Some(position) = position {
branch_rank += position as u64; branch_rank.push(position as u64);
} else { } else {
branch_rank = u64::max_value(); branch_rank.clear();
break; break;
} }
} }
min_rank = min_rank.min(branch_rank);
if !branch_rank.is_empty() {
branch_rank.sort_unstable();
// because several words in same query can't match all a the position 0,
// we substract the word index to the position.
let branch_rank: u64 = branch_rank.into_iter().enumerate().map(|(i, r)| r - i as u64).sum();
// here we do the means of the words of the branch
min_rank = min_rank.min(branch_rank / branch_len as u64);
}
} }
min_rank min_rank