Python3实现写作
导语
T_T没有科研梦想的人半夜过来水篇文章~~~
让Python学会写写歌，创创作~~~
纯属娱乐~~~
改编自PyTorch官网的一个教程，不过我用TF写的，然后生成英文变成了生成中文~~~
Let's Go~~~
相关文件
百度网盘下载链接: https://pan.baidu.com/s/1VUEFR82CqyDj2w9SS4xnFA
密码: 3dmx
开发工具
Python版本：3.6.4
相关模块：tensorflow-gpu模块；numpy模块；以及一些Python自带的模块。
其中TensorFlow-GPU版本为：
1.6.0
环境搭建
安装Python并添加到环境变量，pip安装需要的相关模块即可。
原理简介
关于RNN的介绍，请参考我之前的文章：
Python实现简单的机器翻译模型（点击蓝字进入）
这里我们使用Char RNN模型进行训练。
即Seq2Seq模型中的特殊情况输入与输出等长。
也就是类似下图的样子（图源网络）：
具体实现过程详见相关文件中的源代码。
使用演示
在cmd窗口运行Char_RNN.py文件并根据提示输入相关信息即可。
训练相关的参数请在config.py文件中调整：
训练所用的数据路径以及生成新文本所用的预热数据请在Char_RNN.py文件中调整：
训练演示视频：
因为时间太长了，我只录了训练的前面一部分。
最终训练结果如下图所示：
结果展示
因为时间和资源有限。
这里只展示周杰伦的所有歌词作为样本进行训练之后测试的结果。
当然你完全可以找其他文本作为训练语料来生成风格各异的“文章”，“歌词”等等~~~
结果如下：
（预热歌词为：是曾与你躲过雨的屋檐）
好吧我并不知道它在说啥T_T
一个失败的案例T_T
大概是模型太low了。而且训练数据是我从网上直接下载的，也没有预处理过T_T。中间竟然突然冒出一个括号我也是颓了T_T
上一篇：JVM的内存回收及常见算法
下一篇：设计模式---抽象工厂模式