Pick next word
def pick_word(probabilities, int_to_vocab): | |
word_id = np.argmax(probabilities) | |
word_string = int_to_vocab[word_id] | |
return word_string |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment