Merge pull request #116 from bettersg/fix-reduce-number-of-results

Reduce number of search results to top_k * 3
bettersg · Dec 22, 2024 · bddbb68 · bddbb68
2 parents b4e968f + c75e122
commit bddbb68
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/.gitignore b/.gitignore
@@ -218,3 +218,5 @@ yarn-error.log*
 #dataset_workflow
 node_modules/
 creds.json
+frontend/env.development
+frontend/env.staging
diff --git a/backend/functions/ml_logic/searchModelManager.py b/backend/functions/ml_logic/searchModelManager.py
@@ -357,6 +357,7 @@ def combine_and_aggregate_results(
 
         # Sort by similarity in descending order
         sorted_results = aggregated_results.sort_values(by="Similarity", ascending=False).reset_index(drop=True)
+        sorted_results = sorted_results.head(top_k * 3).reset_index(drop=True)
 
         return sorted_results