Fix result logic

This commit is contained in:
Louis Dureuil 2024-04-11 18:54:03 +02:00
parent 9cef8ec087
commit 5ce4d5f552
No known key found for this signature in database

View File

@ -171,7 +171,14 @@ impl<'a> Recommend<'a> {
let mut document_scores = Vec::with_capacity(self.limit); let mut document_scores = Vec::with_capacity(self.limit);
// skip offset +1 to skip the target document that is normally returned // skip offset +1 to skip the target document that is normally returned
for (docid, distance) in results.into_iter().skip(self.offset + 1) { for (docid, distance) in results.into_iter().skip(self.offset) {
if documents_ids.len() == self.limit {
break;
}
if id == docid {
continue;
}
documents_ids.push(docid); documents_ids.push(docid);
let score = 1.0 - distance; let score = 1.0 - distance;