diff --git a/src/khoj/processor/embeddings.py b/src/khoj/processor/embeddings.py index 1e92f27d..a4daa24f 100644 --- a/src/khoj/processor/embeddings.py +++ b/src/khoj/processor/embeddings.py @@ -27,5 +27,5 @@ class CrossEncoderModel: def predict(self, query, hits: List[SearchResponse]): cross__inp = [[query, hit.additional["compiled"]] for hit in hits] - cross_scores = self.cross_encoder_model.predict(cross__inp) + cross_scores = self.cross_encoder_model.predict(cross__inp, apply_softmax=True) return cross_scores diff --git a/src/khoj/search_type/text_search.py b/src/khoj/search_type/text_search.py index 041c385f..d6f27cea 100644 --- a/src/khoj/search_type/text_search.py +++ b/src/khoj/search_type/text_search.py @@ -224,7 +224,7 @@ def cross_encoder_score(query: str, hits: List[SearchResponse]) -> List[SearchRe # Convert cross-encoder scores to distances and pass in hits for reranking for idx in range(len(cross_scores)): - hits[idx]["cross_score"] = -1 * cross_scores[idx] + hits[idx]["cross_score"] = 1 - cross_scores[idx] return hits