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.
- Downloads last month
- 8
Inference API (serverless) is not available, repository is disabled.