textgenrnn
textgenrnn是一个基于Keras和TensorFlow的Python 3模块,只需几行代码即可训练不同规模和复杂度的文本生成神经网络。支持字符级和词级训练,并可使用预训练模型加快训练。其现代架构利用注意力加权和跳过嵌入等技术,提升模型质量和训练速度。可在GPU上训练并在CPU上生成文本,还可在Colab中免费试用。