IlyaGusev/rubertconv_toxic_clf

Model

15

stars

11

commits

3

repos using this model

2

linked in READMEs

Jul 13, 2022

updated

bert
endpoints_compatible
pytorch
text-classification
text-embeddings-inference
transformers
Browse cluster: RoBERTa Text Classification & NLP

README

RuBERTConv Toxic Classifier

Model description

Based on rubert-base-cased-conversational model

Intended uses & limitations

How to use

Colab: link

from transformers import pipeline

model_name = "IlyaGusev/rubertconv_toxic_clf"
pipe = pipeline("text-classification", model=model_name, tokenizer=model_name, framework="pt") 

text = "Ты придурок из интернета"
pipe([text])

Training data

Datasets:

Augmentations:

  • ё -> е
  • Remove or add "?" or "!"
  • Fix CAPS
  • Concatenate toxic and non-toxic texts
  • Concatenate two non-toxic texts
  • Add toxic words from vocabulary
  • Add typos
  • Mask toxic words with "*", "@", "$"

Training procedure

TBA

Contributors

IlyaGusev

10 commits

system

1 commits

IlyaGusev/rubertconv_toxic_clf

Model

15

stars

11

commits

3

repos using this model

2

linked in READMEs

Jul 13, 2022

updated

bert
endpoints_compatible
pytorch
text-classification
text-embeddings-inference
transformers
Browse cluster: RoBERTa Text Classification & NLP

README

RuBERTConv Toxic Classifier

Model description

Based on rubert-base-cased-conversational model

Intended uses & limitations

How to use

Colab: link

from transformers import pipeline

model_name = "IlyaGusev/rubertconv_toxic_clf"
pipe = pipeline("text-classification", model=model_name, tokenizer=model_name, framework="pt") 

text = "Ты придурок из интернета"
pipe([text])

Training data

Datasets:

Augmentations:

  • ё -> е
  • Remove or add "?" or "!"
  • Fix CAPS
  • Concatenate toxic and non-toxic texts
  • Concatenate two non-toxic texts
  • Add toxic words from vocabulary
  • Add typos
  • Mask toxic words with "*", "@", "$"

Training procedure

TBA

Contributors

IlyaGusev

10 commits

system

1 commits