数据生成

您好,我的任务是直接通过sketch生成文本,我用自己的文本数据训练了模型,但是看到的都是clf和ner的数据增强,请问在哪可以找到对应的数据增强代码可以完成这一任务?