Simple_RNN_Model_predict_next.py

def predict_next(self, sentence: str) -> str:
    # predict the next part of the sentence given as parameter
    a = np.zeros((1, self.a_size))
    for word in sentence.strip().split():
        if word not in vocabulary:
            word = UNK
        x = words2onehot(self.vocab, [word])
        a, y_hat = self(a, x)
    s = ''
    while True:
        word = sample_word(self.vocab, tf.reshape(y_hat, (-1,)))
        if word == EOS:
            break
        s += ' '+word
        x = words2onehot(self.vocab, [word])
        a, y_hat = self(a, x)
    return s