def next_word_topk(self, sentence: str, topK=5):
    self._model.eval()
    return self._predict_next_word_topk(sentence, topK)
def next_word_topk(self, sentence: str, topK=5):
    self._model.eval()
    return self._predict_next_word_topk(sentence, topK)