Change the matcher of Custom QA to Fuzzy matching.

zhanggaoxuan · Dec 6, 2016 · 227a86e · 227a86e
1 parent c72c135
commit 227a86e
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/Chatbot/QuestionAnswering/qaBase.py b/Chatbot/QuestionAnswering/qaBase.py
@@ -15,6 +15,8 @@ def __init__(self):
         self.path = os.path.dirname(__file__)
 
         self.matcher = getMatcher(matcherType="bm25")
+        self.fuzzy_matcher = getMatcher(matcherType="Fuzzy")
+
         self.evaluator = Evaluator()
         self.moduleTest()
 
@@ -69,9 +71,12 @@ def getCustomQA(self, sentence, api_key, threshold=50):
 
         # Load question to a list.
         q_list = [qa["question"] for qa in customqa_list]
-        #TODO  customized threshold.
-        title,index = self.matcher.match(sentence,custom_title=q_list)
-        sim = self.matcher.getSimilarity()
+
+        #TODO 1.  customized threshold.
+
+        #NOTICE: Always choose fuzzy matcher for custom matching.
+        title,index = self.fuzzy_matcher.match(sentence,custom_title=q_list)
+        sim = self.fuzzy_matcher.getSimilarity()
         if sim < threshold:
             return None,0
         return customqa_list[index]["answers"][random.randrange(0,len(customqa_list[index]["answers"]))],sim