Toxic-comment / README.md
vishwapatel123's picture
Upload README.md
1e2c2b1
metadata
license: afl-3.0
language:
  - en
metrics:
  - accuracy
library_name: transformers
pipeline_tag: text-classification

Name

Vishwa Patel

Project

Toxic Comment Classification

Model description

This model is a fine-tuned version of the bert-base-uncased model to classify toxic comments.

Training data

The training data comes from this Kaggle competition. We use 90% of the train.csv data to train the model.