TextBox:TextBox是用于构建文本生成系统的开源库 源码

TextBox(妙笔) “李太白少时,梦所用之笔头上生花后天才赡逸,名闻天下。”-王仁裕《开元天宝遗事·梦笔头生花》 | | | TextBox是基于Python和PyTorch开发的,用于在统一,全面,高效的框架中复制和开发文本生成算法,以用于研究目的。 我们的库包含21种文本生成算法,涵盖两项主要任务: 无条件(无输入)生成 有条件的(Seq2Seq)生成,包括机器翻译,文本摘要,属性到文本和对话系统 我们提供9种基准文本生成数据集的支持。 用户可以应用我们的库来处理原始数据副本,或者简单地由我们的团队下载处理后的数据集。 图片:TextBox的总体架构 特征 统一和模块化的框架。 TextBox建立在PyTorch的基础上,通过将各种模型分离为一组高度可重用的模块,被设计为高度模块化。 全面的模型,基准数据集和标准化评估。 TextBox还包含多种文本生成模型,涵盖基于VA