Skip to content

Instantly share code, notes, and snippets.

@amankharwal

amankharwal/next word.py Secret

Created Jan 19, 2021
Embed
What would you like to do?
import numpy as np
lexicon = {}
def update_lexicon(current : str, next_word : str) -> None:
# Add the input word to the lexicon if it in there yet.
if current not in lexicon:
lexicon.update({current: {next_word: 1} })
return
# Recieve te probabilties of the input word.
options = lexicon[current]
# Check if the output word is in the propability list.
if next_word not in options:
options.update({next_word : 1})
else:
options.update({next_word : options[next_word] + 1})
# Update the lexicon
lexicon[current] = options
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment