textgenrnn是在Keras/Tensorflow基础上搭建的Python包，特性:

  • 有现代的神经网络架构，使用注意力权重和嵌入Embedding来加速训练和提升模型质量
  • 支持字符级别和单词级别的训练
  • 可设置RNN尺寸、RNN层数、是否使用双向RNN
  • 可支持对任意输入文本的训练，包括大文件
  • 可以使用GPU训练，使用CPU生成文本
  • 提供基于GPU的cuDNN，以加速模型训练
  • 使用情景标签训练模型，更快的学习，产出更好的效果。

安装

pip3 install textgenrnn

快速上手

from textgenrnn import textgenrnn

textgen = textgenrnn()
textgen.generate()

Run

[Spoiler] Anyone else find this post and their person that was a little more than I really like the Star Wars in the fire or health and posting a personal house of the 2016 Letter for the game in a report of my backyard.

使用新文本训练新模型也很简单

textgen.train_from_file('hacker_news_2000.txt', num_epochs=1)
textgen.generate()

Run

Project State Project Firefox

生成3个论文标题按照疯狂程度的增加顺序（temperature越高，生成算法偏离学习概率分布的程度越大)

textgen.generate(3, temperature=1.0)

Run

Why we got money “regular alter”

Urburg to Firefox acquires Nelf Multi Shamn

Kubernetes by Google’s Bern


广而告之