diff --git a/Logic/utils.py b/Logic/utils.py index 6c670b4..d9bbc9e 100644 --- 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 index 97d3941..d0069ca 100644 --- 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, )