Shakhovak commited on
Commit
4a15036
·
verified ·
1 Parent(s): ebb1297

Update retrieve_bot.py

Browse files
Files changed (1) hide show
  1. retrieve_bot.py +1 -1
retrieve_bot.py CHANGED
@@ -83,7 +83,7 @@ class ChatBot:
83
  query_encoding,
84
  )
85
  top_scores, top_indexes = top_candidates(
86
- bot_cosine_scores, intent=intent, initial_data=self.scripts, top=10
87
  )
88
  if top_scores[0] < 0.9:
89
  answer = random.choice(low_scoring_list)
 
83
  query_encoding,
84
  )
85
  top_scores, top_indexes = top_candidates(
86
+ bot_cosine_scores, intent=intent, initial_data=self.scripts, top=6
87
  )
88
  if top_scores[0] < 0.9:
89
  answer = random.choice(low_scoring_list)