Chinese_poem_generator 项目使用教程

Chinese_poem_generator 唐诗、宋词生成器，有详细说明 项目地址: https://gitcode.com/gh_mirrors/ch/Chinese_poem_generator

1. 项目目录结构及介绍

Chinese_poem_generator/
├── checkpoints/
│   └── poetrySong/
├── dataset/
│   └── poetrySong/
├── .gitignore
├── .travis.yml
├── LICENSE
├── README.md
├── config.py
├── data.py
├── demo1.png
├── demo2.png
├── evaluate.py
├── main.py
├── model.py
├── panghu1.jpg
├── panghu2.jpg
└── testCase.txt

目录结构介绍

• checkpoints/: 存放训练好的模型检查点文件。
  • poetrySong/: 存放唐诗、宋词生成器的检查点文件。
• dataset/: 存放训练数据集。
  • poetrySong/: 存放唐诗、宋词的数据集文件。
• .gitignore: Git 忽略文件配置。
• .travis.yml: Travis CI 配置文件。
• LICENSE: 项目许可证文件。
• README.md: 项目说明文档。
• config.py: 项目配置文件。
• data.py: 数据处理脚本。
• demo1.png: 示例图片1。
• demo2.png: 示例图片2。
• evaluate.py: 模型评估脚本。
• main.py: 项目主启动文件。
• model.py: 模型定义脚本。
• panghu1.jpg: 示例图片3。
• panghu2.jpg: 示例图片4。
• testCase.txt: 测试用例文件。

2. 项目启动文件介绍

main.py

main.py 是项目的启动文件，负责控制整个项目的运行流程。通过命令行参数可以指定不同的操作模式，如训练、测试和生成藏头诗。

使用方法

python3 main.py -m [train | test | head]

• train: 训练模式，用于训练模型。
• test: 测试模式，用于随机生成诗歌。
• head: 藏头诗模式，用于生成藏头诗。

3. 项目配置文件介绍

config.py

config.py 是项目的配置文件，包含了项目运行所需的各种参数设置。

主要配置项

• type: 指定训练数据的类型，如 poetrySong 表示唐诗、宋词。
• dataset_path: 数据集路径。
• checkpoint_path: 检查点文件路径。
• epochs: 训练轮数。
• batch_size: 批处理大小。

示例配置

type = 'poetrySong'
dataset_path = './dataset/poetrySong/poetrySong.txt'
checkpoint_path = './checkpoints/poetrySong/checkpoint'
epochs = 10
batch_size = 32

通过修改 config.py 中的配置项，可以调整项目的运行参数，以适应不同的需求。

Chinese_poem_generator 唐诗、宋词生成器，有详细说明 项目地址: https://gitcode.com/gh_mirrors/ch/Chinese_poem_generator