简易lstm写诗机文章

monteCarloStyle 2018-08-08 14:58

description 1.模型生存环境 系统环境:Ubuntu16.04L; 解释器:anaconda(python3), tensorflow-gpu; 硬件支持:i7-7700, gtx-1060; 2. 神经网络结构 隐层由3层lstm单元(3 * 128)构成,输出层由softmax构成。使用加权交叉熵计算损失。 3.模型运行。 初次运行模型需要在目录下建立rnn_log文件夹作为模型保存目录。 1.在model.py中更改模型参数。默认每层神经元128,隐层为3层,学习率为0.001。 2.目录下运行train.py训练模型 3.运行writer.py写诗 4.运行结果


模型
评论
支持Markdown和数学公式,公式格式:\\(...\\)或\\[...\\]

还没有评论