Downloads · 30 days
9.8K
17% of all-time downloads
gector-roberta-base-5k
gotutiyan/gector-roberta-base-5k
gector-roberta-base-5k is a machine learning model from gotutiyan. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for transformers.
All-time downloads
59K
Public
Parameters
128M
1 GB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin512 MB · 50%
Parameter types
How the weights are stored.
F32128M · 100%
Model card
From the Hugging Face model README
Only non-commercial purposes.
gector sample
This is an unofficial pretrained model of GECToR (Omelianchuk+ 2020).
How to use
The code is avaliable from https://github.com/gotutiyan/gector.
CLI
python predict.py --input <raw text file> --restore_dir gotutiyan/gector-roberta-base-5k --out <path to output file>
API
from transformers import AutoTokenizer
from gector.modeling import GECToR
from gector.predict import predict, load_verb_dict
import torch
model_id = 'gotutiyan/gector-roberta-base-5k'
model = GECToR.from_pretrained(model_id)
if torch.cuda.is_available():
model.cuda()
tokenizer = AutoTokenizer.from_pretrained(model_id)
encode, decode = load_verb_dict('data/verb-form-vocab.txt')
srcs = [
'This is a correct sentence.',
'This are a wrong sentences'
]
corrected = predict(
model, tokenizer, srcs,
encode, decode,
keep_confidence=0.0,
min_error_prob=0.0,
n_iteration=5,
batch_size=2,
)
print(corrected)
- Library
- transformers
- Languages
- en
- Created
- Aug 21, 2023
- Updated
- Mar 7, 2026