search

susanli2016 / NLP-with-Python

Scikit-Learn, NLTK, Spacy, Gensim, Textblob and more
2.72k stars 2.01k forks source link

Text generation based on next word prediction #37

Open fawern opened 6 months ago

fawern commented 6 months ago

I created a simple text generation model based on next word prediction with LSTM using the Warhammer 40k novels dataset. The model is trained on the dataset and generates the next word based on the input sentence.