Skip to content

Latest commit

 

History

History
67 lines (52 loc) · 2.1 KB

README.md

File metadata and controls

67 lines (52 loc) · 2.1 KB

LLaMA for Chinese Spelling Correction

简介

中文文本纠错任务是一项NLP基础任务,其输入是一个可能含有语法错误的中文句子,输出是一个正确的中文句子。 语法错误类型很多,有多字、少字、错别字等,目前最常见的错误类型是错别字。大部分研究工作围绕错别字这一类型进行研究。 本项目基于LLaMA实现了中文拼写纠错和语法纠错。

Usage

安装依赖项

  • loguru
  • transformers>=4.33.2
  • datasets
  • tqdm>=4.47.0
  • accelerate>=0.21.0
  • peft>=0.5.0

运行命令:

pip install transformers peft -U

快速加载

pycorrector调用预测

example: examples/gpt/demo.py

from pycorrector import GptCorrector
m = GptCorrector()
print(m.correct_batch(['今天新情很好', '你找到你最喜欢的工作,我也很高心。']))

output:

[{'source': '今天新情很好', 'target': '今天心情很好', 'errors': [('', '', 2)]},
{'source': '你找到你最喜欢的工作,我也很高心。', 'target': '你找到你最喜欢的工作,我也很高兴。', 'errors': [('', '', 15)]}]

Dataset

toy data

中文语法纠错数据(1k条):examples/data/grammar/train_sharegpt.jsonl

data format:

{"conversations":[{"from":"human","value":"对这个句子语法纠错\n\n这件事对我们大家当时震动很大。"},{"from":"gpt","value":"这件事当时对我们大家震动很大。"}]}

big train data

Train model

run train:

cd examples/gpt
python train_chatglm_demo.py --do_train --do_predict

output:

input  : 这块名表带带相传
predict: 这块名表代代相传