poem_robot

本项目是基于transformers库构建的对诗模型。受启发于另外一个使用seq2seq模型构建的单轮对话模型。

目前模型的效果一般，待进一步完善...

项目结构

│  .gitignore
│  README.md
│  requirements.txt
│  train.py
│  main.py
│
├─data
│      poem.txt
│
├─finetuned_models

环境配置

环境配置需要两个步骤：

• 下载项目文件
• 安装相关的第三方库

具体步骤如下：

[email protected]:theonlybigmiao/poem_robot.git && cd poem_robot
pip install requirements.txt

模型使用

• 模型训练

python train.py

• 模型预测

训练好的模型会保存在finetuned_models目录下。

• 体验交互式对诗

python main.py

注意

当前仅支持中文对诗，输入英文会出错。如果使用vs code且遇到对诗页面加载不出来的情况，请转到PyCharm运行。