IDEA-CCNL / GTS-Engine

GTS Engine: A powerful NLU Training System。GTS引擎(GTS-Engine)是一款开箱即用且性能强大的自然语言理解引擎,聚焦于小样本任务,能够仅用小样本就能自动化生产NLP模型。
https://gtsfactory.com/
Apache License 2.0
87 stars 9 forks source link

Dev liuyibo #70

Open rookielyb opened 1 year ago

rookielyb commented 1 year ago

1.增加Keyphrase Generation项目
2.在examples/text_generation/ 新增 Keyphrase Generation的启动项 以及少量数据 3.新增gts_engine/pipelines/qiankunding_generation.py 主代码 3.新增gts_engine/qiankunding/dataloaders/text_generation/dataloader_kgt5.py的数据处理代码 4.新增gts_engine/qiankunding/models/text_generation/t5_kg.py 模型代码 5.在gts_engine/qiankunding/utils/evaluation.py 中新增 TextGenerateEvaluator 类 用于Generate的评估 6.在gts_engine/qiankunding/utils/tokenization.py中新增 T5的tokenization 7.运行examples/text_generation/run_train_qiankunding.sh 成功如下

企业微信截图_6115ece4-cd12-42f3-bb03-2287ae72ec5b

8.运行examples/text_generation/run_inference_qiankunding.sh 成功如下

企业微信截图_928e3f9a-dbc0-4fe5-ac39-bdc49ec50a8c