Skip to content

📝基于RNN的中国古诗词生成模型(SJTU CS382 Course Project)

Notifications You must be signed in to change notification settings

Forcrush/Chinese_Poem_Generator

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 

Repository files navigation

基于 RNN 的中国古诗词生成模型

运行环境

  • Windows 8.1 64 位系统

  • Python 3.5.x ( 64 位)

  • PyCharm 2018.1.3

项目结构

该项目是使用 python 语言,使用基于 Django 开发的 Web 界面,主要结构如下:

  • dataset 文件夹,数据训练集,包含唐诗、宋词和诗经
  • log 文件夹,存储训练好的模型
  • RNN_poem_gen 包,基于 Django 框架的主要业务逻辑与后端处理
  • poem_generator 包,项目的相关设置
  • static 文件夹,网页端使用到的 css 样式和一些 js 代码,用于渲染网页
  • templates 文件夹,html 网页代码,通过 Django 注入数据,作为前端 UI
  • db.sqlite3,Django 默认的一个数据库设置
  • manage.py,Django Web 运行的入口
  • poetry.py,包含模型构建,数据处理,训练和测试调度接口

训练和测试

  1. 若要重新训练模型,进入 poetry.py 运行 begin_train() 可指定训练神经单元 Basic_RNNLSTMGRU

  2. 训练完成后可运行 begin_gen(),其中默认参数为空字符串,即随机生成诗词;也可指定一段文字生成藏头诗

程序运行方式

可以通过 PyCharm 打开(两种方式):

  • 直接运行 Django 项目

  • 运行 manage.py,主要运行时需要将参数设置为 runserver 8000

通过命令行运行:

  • cdmanage.py 目录下,执行指令 python manage.py runserver 8000

Web UI 演示运行

上述两种方式运行完毕后,若出现如下的提示,可以访问 http://127.0.0.1:8000 ,看到项目的 UI 界面,根据内容直接在网页上操作即可

December 27, 2018 - 15:03:30
Django version 2.1.4, using settings 'poem_generator.settings'
Starting development server at http://127.0.0.1:8000/
Quit the server with CTRL-BREAK.

About

📝基于RNN的中国古诗词生成模型(SJTU CS382 Course Project)

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published