- Kaggle Challenge & Dataset
- Heavily inspired from this Kaggle Notebook
The model is trained in toxic_comment_classification.ipynb using a pretrained BERT model from HuggingFace. The latest model and details are uploaded to S3. The testing and report generations is found in model_test.ipynb. model_test.ipynb will download the latest version of the model trained in toxic_comment_classification.ipynb.
- Both notebooks should run end to end with provided inline instructions.
Latest Generated Report is found here.