1. 首页
  2. 人工智能
  3. 深度学习
  4. SKchat小说续写模型,深度学习应用于文学创作

SKchat小说续写模型,深度学习应用于文学创作

上传者: 2023-11-28 08:04:29上传 ZIP文件 425.45MB 热度 60次

SKchat-v1续写模型是基于深度学习技术的一种创新应用,经过4个epochs的训练,该模型采用了英文小说数据进行微型续写。其代码框架由run.py构成,train.txt包含了用于训练的英文小说数据。模型文件夹内存放着训练后的模型,其中每套模型都包括三个hdf5文件,分别是a.hdf5、a_en.hdf5和a_de.hdf5。在模型训练过程中,testchat.txt则用于自动测试文本的性能。需要注意的是,输入的句子最长不超过90个字符,超过部分会被自动截断。建议用户自行进行训练,因为自带的模型效果并不理想。

下载地址
用户评论