Next word prediction pytorch. tokenize import RegexpTokenizer from keras.