[发明专利]基于FocalGAN的短文本自动生成方法、装置、设备及存储介质

权利要求书

1.一种基于Focal GAN的短文本自动生成方法，其特征在于，所述方法包括：

构建基于端到端的生成器及基于焦点损失函数的判别器；

根据所述生成器和判别器生成对抗网络；

将目标单词作为所述对抗网络的输入序列，以使得根据所述对抗网络输出所述目标单词对应的短文本。

2.如权利要求1所述的方法，其特征在于，所述构建基于端到端的生成器及基于焦点损失函数的判别器之前，所述方法还包括：

获取语料集，所述语料集至少包括作文语料、头条新闻语料；

对所述语料集进行预处理得到筛选后的语料集；

根据预设划分比例将所述筛选后的语料集划分为训练集、测试集、验证集。

3.如权利要求2所述的方法，其特征在于，在所述将目标单词作为所述对抗网络的输入序列，以使得根据所述对抗网络输出所述目标单词对应的短文本之前，所述方法还包括：

根据所述训练集对所述对抗网络进行训练。

4.如权利要求2所述的方法，其特征在于，所述预设划分比例为6：2:2。

5.如权利要求3所述的方法，其特征在于，所述根据所述训练集对所述对抗网络进行训练，包括：

从所述训练集采样得到监督训练所需要的正例；

根据所述监督训练所需要的正例对所述对抗网络进行训练。

6.如权利要求1所述的方法，其特征在于，在所述根据所述生成器和判别器生成对抗网络之后，所述将目标单词作为所述对抗网络的输入序列，以使得根据所述对抗网络输出所述目标单词对应的短文本之前，所述方法还包括：

基于注意力机制及所述判别器反馈给所述生成器的奖励修正所述生成器的分配权重。

7.如权利要求1所述的方法，其特征在于，所述对所述语料集进行预处理得到筛选后的语料集，包括：

对所述语料集进行去除重复内容、分词、去除噪声、选取字数长度均匀中的至少一项操作。

8.一种基于FocalGAN的短文本自动生成装置，其特征在于，所述装置包括：

构建模块，用于构建基于端到端的生成器及基于焦点损失函数的判别器；

生成模块，用于根据所述生成器和判别器生成对抗网络；

输入模块，用于将目标单词作为所述对抗网络的输入序列，以使得根据所述对抗网络输出所述目标单词对应的短文本。

9.一种基于FocalGAN的短文本自动生成设备，其特征在于，所述设备包括：

存储有可执行程序代码的存储器；

与所述存储器耦合的处理器；

所述处理器调用所述存储器中存储的所述可执行程序代码，执行如权利要求1-7任一项所述的基于FocalGAN的短文本自动生成方法。

10.一种计算机存储介质，其特征在于，所述计算机存储介质存储有计算机指令，所述计算机指令被调用时，用于执行如权利要求1-7任一项所述的基于FocalGAN的短文本自动生成方法。