更详细的使用方法参见具体文件夹下的README.md,这里只是简单介绍一下各个文件夹的作用。
我们自己搭建的lstm/gru模型
框架:torch
我们自己搭建的transformer模型
框架:torch
我们自己搭建的lstm模型,并在其基础上利用gradio搭建了一个简单的网页应用
hugging face上的demo:https://zhangj726-poetrygeneration.hf.space/
框架:torch
代码来源于https://github.com/920232796/bert_seq2seq
我们利用了其中的Roberta模型,进行了修改,使其可以用于我们的任务
框架:torch
代码来源于https://github.com/AaronJny/DeepLearningExamples/blob/master/keras-bert-poetry-generator
我们利用了其中的bert和NEZHA模型,进行了修改,使其可以用于我们的任务
框架:keras