Merge pull request #8 from Moeen89/main

fix argument of utils.search()
sharif-ml-lab · May 31, 2024 · e9835d3 · e9835d3
2 parents 9fe8b77 + 9d77a95
commit e9835d3
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/Logic/utils.py b/Logic/utils.py
@@ -38,6 +38,9 @@ def search(
     weights: list = [0.3, 0.3, 0.4],
     should_print=False,
     preferred_genre: str = None,
+    smoothing_method = None, 
+    alpha=0.5, 
+    lamda=0.5, 
 ):
     """
     Finds relevant documents to query

diff --git a/UI/main.py b/UI/main.py
@@ -144,7 +144,7 @@ def search_handling(
                 search_max_num,
                 search_method,
                 search_weights,
-                unigram_smoothing=unigram_smoothing,
+                smoothing_method = unigram_smoothing,
                 alpha=alpha,
                 lamda=lamda,
             )