Spaces:

Shakhovak
/

Sheldon_Retrieval_chat_bot

Sleeping

Shakhovak commited on Feb 11

Commit

69f2a1b

•

1 Parent(s): ccc5364

Update retrieve_bot.py

Files changed (1) hide show

retrieve_bot.py CHANGED Viewed

@@ -44,7 +44,7 @@ class ChatBot:
             utterance, self.ranking_model, contexts=self.conversation_history
         )
         bot_cosine_scores = cosine_sim(self.vect_data, query_encoding)
-        top_scores, top_indexes = top_candidates(bot_cosine_scores, top=10)
         # test candidates and collects them with label 0 to dictionary

             utterance, self.ranking_model, contexts=self.conversation_history
         )
         bot_cosine_scores = cosine_sim(self.vect_data, query_encoding)
+        top_scores, top_indexes = top_candidates(bot_cosine_scores, top=5)
         # test candidates and collects them with label 0 to dictionary