Grammar T5

🚀 語法糾錯模型

這是一個用於語法糾錯的模型，使用了 jfleg 數據集，並基於 t5-base 構建。雖然僅訓練了 3 個輪次，輸出效果可能有限，但它為語法糾錯任務提供了一個基礎的解決方案。

🚀 快速開始

本模型可結合標準的 transformers 庫使用，且模型規模較小，能夠在 CPU 上運行。

📦 安裝指南

在使用該模型之前，你需要安裝相關依賴：

$ pip install transformers torch sentencepiece

💻 使用示例

基礎用法

安裝好依賴後，你就可以運行以下代碼來使用該模型：

from transformers import AutoModelForSeq2SeqLM, AutoTokenizer

model_name = 'vagmi/grammar-t5'

model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
tokenizer = AutoTokenizer.from_pretrained(model_name)

text = 'fix grammar: I am work with machine to write gooder english.'
inputs = tokenizer(text, return_tensors='pt')
outputs = model.generate(inputs['input_ids'], num_beams=2, max_length=512, early_stopping=True)
fixed = tokenizer.decode(outputs[0], skip_special_tokens=True)
# I am working with machine to write better english.

📄 許可證

本項目採用 Apache-2.0 許可證。