T5 Base Grammar Correction

🚀 T5语法纠错模型

本模型旨在生成输入文本的修订版本，以减少其中的语法错误。它借助特定数据集进行训练，能有效提升文本的语法准确性，为用户提供高质量的文本修正服务。

🚀 快速开始

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)

# Add the prefix "grammar: " before each input 
result = happy_tt.generate_text("grammar: This sentences has has bads grammar.", args=args)

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

📄 许可证

本项目采用CC BY-NC-SA 4.0许可协议。