Skip to content

Instantly share code, notes, and snippets.

@chricke

chricke/pick_word.py

Created Mar 26, 2019
Embed
What would you like to do?
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
You can’t perform that action at this time.