Przeglądaj źródła

Return top 5 results

rota3015 7 miesięcy temu
rodzic
commit
33dd85efbd
1 zmienionych plików z 1 dodań i 1 usunięć
  1. 1 1
      chatbot/routes.py

+ 1 - 1
chatbot/routes.py

@@ -80,7 +80,7 @@ def find_closest_match(user_input):
         results.append((max_similarity_curr, return_value, ruleId))
         # print(results)
     results.sort(reverse=True, key=lambda x: x[0])
-    return results[:3]
+    return results[:5]
 
 def similarity_score(set1, set2):
     # Calculate the intersection of the two sets