hunkim / word-rnn-tensorflow

Multi-layer Recurrent Neural Networks (LSTM, RNN) for word-level language models in Python using TensorFlow.
MIT License
1.3k stars 494 forks source link

How to predict the best word from set of words? #83

Open AustinZuniga opened 5 years ago

AustinZuniga commented 5 years ago

I'm trying to use your Word level RNN to predict the best word from a set of words. let's say the set of words will be: set_of_words = {beautiful, beauty, bea}

and the output of the rnn model = "beautiful"

how can i do this using your rnn model?