语言: 中文
标签:

• 语法
• 文本生成
  许可证: cc-by-nc-sa-4.0
  数据集:
• jfleg

T5语法校正模型

该模型可生成输入文本的修订版本，旨在减少语法错误。
它通过Happy Transformer框架训练完成，使用的数据集为JFLEG。此处提供完整教程介绍如何训练类似模型。

使用方法

pip install happytransformer

from happytransformer import HappyTextToText, TTSettings  

happy_tt = HappyTextToText("T5", "vennify/t5-base-grammar-correction")  

args = TTSettings(num_beams=5, min_length=1)  

# 在每段输入前添加前缀"grammar: "  
result = happy_tt.generate_text("grammar: This sentences has has bads grammar.", args=args)  

print(result.text) # 输出：This sentence has bad grammar.