Model Details of roberta_first_toxicity_classifier
Toxicity Classification Model (but for the first part of the data)
This model is trained for toxicity classification task. The dataset used for training is the merge of the English parts of the three datasets by
Jigsaw
(
Jigsaw 2018
,
Jigsaw 2019
,
Jigsaw 2020
), containing around 2 million examples. We split it into two parts and fine-tune a RoBERTa model (
RoBERTa: A Robustly Optimized BERT Pretraining Approach
) on it. THIS MODEL WAS FINE-TUNED ON THE FIRST PART. The classifiers perform closely on the test set of the first Jigsaw competition, reaching the
AUC-ROC
of 0.98 and
F1-score
of 0.76.
How to use
from transformers import RobertaTokenizer, RobertaForSequenceClassification
# load tokenizer and model weights, but be careful, here we need to use auth token
tokenizer = RobertaTokenizer.from_pretrained('SkolkovoInstitute/roberta_toxicity_classifier', use_auth_token=True)
model = RobertaForSequenceClassification.from_pretrained('SkolkovoInstitute/roberta_toxicity_classifier', use_auth_token=True)
# prepare the input
batch = tokenizer.encode('you are amazing', return_tensors='pt')
# inference
model(batch)
roberta_first_toxicity_classifier huggingface.co is an AI model on huggingface.co that provides roberta_first_toxicity_classifier's model effect (), which can be used instantly with this s-nlp roberta_first_toxicity_classifier model. huggingface.co supports a free trial of the roberta_first_toxicity_classifier model, and also provides paid use of the roberta_first_toxicity_classifier. Support call roberta_first_toxicity_classifier model through api, including Node.js, Python, http.
roberta_first_toxicity_classifier huggingface.co is an online trial and call api platform, which integrates roberta_first_toxicity_classifier's modeling effects, including api services, and provides a free online trial of roberta_first_toxicity_classifier, you can try roberta_first_toxicity_classifier online for free by clicking the link below.
s-nlp roberta_first_toxicity_classifier online free url in huggingface.co:
roberta_first_toxicity_classifier is an open source model from GitHub that offers a free installation service, and any user can find roberta_first_toxicity_classifier on GitHub to install. At the same time, huggingface.co provides the effect of roberta_first_toxicity_classifier install, users can directly use roberta_first_toxicity_classifier installed effect in huggingface.co for debugging and trial. It also supports api for free installation.
roberta_first_toxicity_classifier install url in huggingface.co: