Neural network language model in python