kingeorge / nlp-final

NLP Final Assignment Source Code

更详细的使用方法参见具体文件夹下的README.md，这里只是简单介绍一下各个文件夹的作用。

lstm-gru

我们自己搭建的lstm/gru模型

框架：torch

transfomer

我们自己搭建的transformer模型

框架：torch

lstm-bySong-with-gradio

我们自己搭建的lstm模型，并在其基础上利用gradio搭建了一个简单的网页应用

hugging face上的demo：https://zhangj726-poetrygeneration.hf.space/

框架：torch

bert_seq2seq

代码来源于https://github.com/920232796/bert_seq2seq

我们利用了其中的Roberta模型，进行了修改，使其可以用于我们的任务

框架：torch

keras-bert-poetry-generator

代码来源于https://github.com/AaronJny/DeepLearningExamples/blob/master/keras-bert-poetry-generator

我们利用了其中的bert和NEZHA模型，进行了修改，使其可以用于我们的任务

框架：keras