junxi的写诗机文章

junxi 2018-08-14 00:03

模型使用了RNN、LSTM模型,训练了502390 个step,结果中只上传了是用到的ckpt文件

整个训练及写诗的命令: 1、生成embedding及字典:python word2vec_chi.py

2、训练命令:python ./train.py --output_dir=./ckpt --text=./QuanSongCi.txt --num_steps=32 --batch_size=16 --learning_rate=0.001 --dictionary=./dictionary.json --reverse_dictionary=./reverse_dictionary.json --embedding_file=./embedding.npy

3、写诗命令:python sample.py

从输出的文字里分别截取一段,形成五首诗:

第一首:《思》 肠与愁相人,音月深不红。 江水不有来,月月风上寒。 念心东风上,人行入寄情。 时生愁上影,一到自长路。

第二首 《念》 别人一人去,长谁故相家。 忆心旧复行,人在人人路。 人去远远在,书战可远别。 早心怀有人,愁望江上门。

第三首:《恋》 黄云在云远,不心与相春。 此日水自日,泪处临门秋。 不知自来远,不心相相忆。 秋江深风绿,人人离人里。

第四首《花城初夜》青风春山无秋风,江风飞山相相人。 无见初夜风花时,花城深月天花听。

第五首:《情千》 春时何见初风在,寒声北时海叶声。 无门上明空花水,白叶飞远万难人。 白月风在几花云,愁罗城落春前深。 天风入成空不处,春在风水还不时。


模型
框架:TensorFlow
{{panelTitle}}(2)
支持Markdown和数学公式,公式格式:\\(...\\)或\\[...\\]
西晽 2018-10-16 19:52

这些诗都不顺句啊

junxi 2018-08-14 00:09

补充一下:运行环境tensorflow 1.5

结果截图: description

description

关注微信公众号