Recurrent neural network text generation

This project trains a recurrent neural network (RNN) that generates text imitating the style of Kalevala and Kanteletar.

A character level RNN is implemented using Keras. The trained model is a stateful RNN with two LSTM layers.

Usage

Prepare the environment:

npm install
pipenv install
pipenv shell

Training:

python -m src.preprocess
python -m src.train
python -m src.convert2tfjs "weights/2020-07-07T21:19:37/"  # <- replace the path!

Generating text from a trained model:

python -m src.sample -n 1000 --temperature 0.4 --preseed "Sanoi vanha Väinämöinen" weights/<path>

Run the app

npm run start

Deploying

npm run zip

Upload saesampo.zip to AWS Amplify.

License

MIT license

The texts of Kalevala and Kanteletar included in the data subdirectory are in public domain.