def next_word(self, sentence: str, next_word: str):
    self._model.eval()
    temp_str = [x for x in light_tokenize(sentence)]
    predict_prob = self._predict_next_word_prob(temp_str, next_word)
    return predict_prob.item()
def next_word(self, sentence: str, next_word: str):
    self._model.eval()
    temp_str = [x for x in light_tokenize(sentence)]
    predict_prob = self._predict_next_word_prob(temp_str, next_word)
    return predict_prob.item()