Downloads · 30 days
13
2% of all-time downloads
bert-base-finetuned-toxic-comment-classification
ZiruiXiong/bert-base-finetuned-toxic-comment-classification
bert-base-finetuned-toxic-comment-classification is a text classification model from ZiruiXiong. Use it when you need a label for a piece of text. It is set up for transformers. The card lists the license as afl-3.0.
This model is a fine-tuned version of bert-base-uncased on the Kaggle jigsaw toxic comment classification dataset.
All-time downloads
720
Public
Repo size
1.3 GB
Likes
3
Public
Repo makeup
Click a slice to open those files.
.bin438 MB · 100%
At a glance
- Task
- Text Classification
- Library
- transformers
- License
- afl-3.0
- Model type
- bert
- Access
- Public
- Created
- Mar 11, 2023
- Updated
- Mar 28, 2023
- SHA
- bedc0437
Try a prompt
Model card
From the Hugging Face model README
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.
- Task
- Text Classification
- Library
- transformers
- Type
- bert
- License
- afl-3.0
- Languages
- en
- Created
- Mar 11, 2023
- Updated
- Mar 28, 2023