Zmiana test_nlu

This commit is contained in:
Anna Nowak 2021-05-16 21:49:53 +02:00
parent 1af80df028
commit 3c39ab377d

View File

@ -104,7 +104,7 @@ class ML_NLU:
def test_nlu(self, utterance):
if utterance:
return tabulate(self.predict(self.model, 'doktor lekarza rodzinnego najlepiej dzisiaj w godzinach popołudniowych dziś '.split()), tablefmt='html')
return tabulate(self.predict(self.model, utterance.split()), tablefmt='tsv')
else:
return 'Critical Error'