Skip to content
🎊 Automatically write Chinese ancient poems | 自动写古诗
Python
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
AutoPoem
dataset
.gitignore
First_sentence.py
First_word.py
LICENSE
README.MD
Random.py
Tibetan_poetry.py
config.py
poetry_model.png
train.py

README.MD

AutoPoem 自动古诗生成器

本项目使用这些技术:

  • RNN
  • LSTM
  • Keras
  • Tensorflow

输出结果

策杖政成时,山无火不白。若生心中心,可为子我间。
侧闻郡守至,入事天新尚。非长吟风病,何过远城不。
候馆扫清昼,静日君心缘。发游心休心,处雪白成间。
晚景微雨歇,归门深长鸟。心间老望回,头柳空月下。
亩浍敷四海,心野无未微。时尚月初行,自不夕闻古。
关河月未晓,雪中人终后。肠异知谁名,堪薄难今生。

环境配置

模型运行在Python3下,请注意.

pip install tensorflow Keras h5py numpy

# 如果使用GPU,请使用以下命令
conda install keras-gpu

调用方法

先进行设置

class Config(object):
    poetry_file = 'dataset/poetry.txt'   # 数据集
    weight_file = 'poetry_model.h5'      # 模型
    max_len = 6                          # 根据前六个字预测第七个字
    batch_size = 128                     # Batch Size
    learning_rate = 0.0005               # Learning Rate

写藏头诗

from AutoPoem.Model import *
from config import Config


if __name__ == '__main__':
    model = PoetryModel(Config)
    for i in range(3):
        sen = model.predict_hide('我不要你')
        print(sen)

# 输出:我何人清落,不后百能天。要作欲清自,你知自人九。

随机写诗

from AutoPoem.Model import *
from config import Config


if __name__ == '__main__':
    model = PoetryModel(Config)
    for temp in [0.5, 1, 1.5]:
        sen = model.predict_random(temperature=temp)
        print(sen)
        
# 输出:名士竹林隈,人池生行是。事无松有所,有相此门人。

模型:

参考了:

You can’t perform that action at this time.