SKchat-v1续写模型是一款基于微型深度学习技术的创新应用,特别设计用于英文小说的续写。通过经过4个epochs的训练,该模型在run.py代码框架的支持下,利用train.txt中的英文小说数据进行模型训练。训练后的模型保存在model文件夹中,每套模型包含a.hdf5、a_en.hdf5和a_de.hdf5三个文件。测试时使用的文本为testchat.txt。需要注意的是,用户输入的句子最长不能超过90个字符,超过的部分会被截断。值得指出的是,自带的模型效果并不理想,因此建议用户自己进行模型训练,以达到更好的续写效果。
暂无评论