#116 · Primary category: NLP Tools & Text Processing

gector

bert grammatical-error-correction natural-language-processing nlp roberta sequence-labeling text-simplification transformers xlnet

Official implementation of the papers "GECToR – Grammatical Error Correction: Tag, Not Rewrite" (BEA-20) and "Text Simplification by Tagging" (BEA-21)

Project last updated:05/21/24

GitHub Stars

972

Forks

220

Contributors

7

License

Apache-2.0

Why we included this project

If you're building grammar checking into an editor or writing tool, GECToR is worth studying. It's the official PyTorch code for a tagging-based approach to grammatical error correction: instead of rewriting the whole sentence, the model assigns each token a small edit, which keeps inference cheap enough to run at scale. The repo includes a full training pipeline and pretrained checkpoints for BERT, RoBERTa, and XLNet that score well on standard GEC benchmarks, so you can load those weights for immediate inference or fine-tune them on your own text. It also ships the related text-simplification-by-tagging model behind the same interface, covering two editing tasks from one codebase. The preprocessing scripts and the two-stage fine-tuning recipe give you a concrete baseline to compare against your data before you bet on a production writing-assistant stack.

Articles for this project

No articles for this project yet.

To suggest a topic or contribute an article, contact us.

Related projects in this category