Name		Name	Last commit message	Last commit date
Latest commit History 9 Commits
archive		archive
img		img
.gitignore		.gitignore
dataset.py		dataset.py
main.py		main.py
model.py		model.py
readme.md		readme.md

Repository files navigation

Poetry Generate

基于Transformer的诗句生成模型，效果不算完美。这是训练116次之后的效果

Usage

有两种生成模式

直接生成一首诗，输入风格输出整首诗
输入每一句诗的开头，输出藏头诗

Data Source

数据来源于ChinesePoetryDataset，训练只采用了前5W行。

About

古诗生成模型，基于Transformer/Chinese poetry geneation, based on transformer.

pytorch transformer-pytorch

Report repository

Releases

No releases published

Packages

No packages published

Languages

Python 100.0%