ZiruiXiong/bert-base-finetuned-toxic-comment-classification

bert-base-cased-finetuned-jigsaw

This model is a fine-tuned version of bert-base-uncased on the Kaggle jigsaw toxic comment classification dataset.

How to use

You can use the model with the following code.

from transformers import AutoModelForSequenceClassification, AutoTokenizer, TextClassificationPipeline

model_path = "ZiruiXiong/bert-base-finetuned-toxic-comment-classification"
tokenizer = AutoTokenizer.from_pretrained(model_path)
model = AutoModelForSequenceClassification.from_pretrained(model_path)

pipeline = TextClassificationPipeline(model=model, tokenizer=tokenizer)
print(pipeline('You are also a liar since I hardly spoke with you.'))

Evaluation results

The model achieves 0.95 AUC in a 1500 rows held-out test set.