mumufei / ai_writer

该项目可以根据用户给出的上文自动生成下文 该项目是本人的本科毕业设计。项目主要基于GPT-2 Chinese实现。本人的工作主要是用新的语料库进行了几次训练，得出来了一个还凑合的模型。该项目已经初步完成，不再进行进一步的更新。

演示视频

项目的深度学习代码基于GPT-2 Chinese项目。 安装方法： 请先于NVIDIA官网下载

0. 显卡驱动
1. CUDA 10.1
2. cuDNN 7.6.5 for CUDA 10.1

之后，在Windows Terminal中将工作路径跳转到gpt-2 chinese目录下输入pip install -r requirements.txt

本人在几乎对GPT-2没有改动的前提下，为该项目添加了bat脚本：

0. train.bat 训练脚本
1. generate.bat 生成脚本

可以根据自己的需要调整bat脚本

-语料库-

如果您想使用我们的语料库，可以在网盘链接（提取码：9k3e）下载语料库。语料库包含：

0. 已经转换为GPT-2 Chinese要求的JSON格式的《最终幻想14》截至5.1版本的剧本 1） 已经转换为GPT-2 Chinese要求的JSON格式的《少女前线》截至4月1日的剧本 2） 尚未转换为GPT-2 Chinese要求的JSON格式的《Fate/Grand Order》1.0的剧本

爬取和整合数据的方式可以参考文件夹“Spider”中的Python脚本

-模 型-

本人使用了散文语料预模型、百度贴吧语料预模型训练了一些可以生成《最终幻想14》风格剧本和《少女前线》风格剧本的模型，如果有需要可以通过网盘链接（提取码：uu7p）下载。