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