minimaxir/textgenrnn
A Python module for training character-level and word-level RNNs to generate text on any dataset.

Velocity · 7d
+1.5
★ / day
Trend
→steady
star history
textgenrnn is a Keras/TensorFlow wrapper that simplifies training recurrent neural networks for text generation. It supports both character-level and word-level generation, configurable RNN sizes and layers, bidirectional processing, and GPU acceleration via CuDNN. The library implements attention-weighting and skip-embedding techniques to improve training speed and output quality.