Chrome extension which detects, classifies and highlights toxic comments under Youtube videos
There are three parts in the application:
-
- toxic_lstm_with_identities.ipynb - LSTM model which also takes into the account mentioned identites (main model for this project!)
- youtube_api.ipynb - shows how to use youtube API in python
- toxic_lstm.ipynb - LSTM model training
- inference.ipynb - inference
- bert.ipynb - BERT training (work in progress)
The model is built using the dataset from recent Jigsaw Kaggle competition.
-
- Go to chrome://extensions
- Enable Developer mode by toggling the button in the upper-right corner
- Use the 'Load unpacked extension' option and select the folder, containing the extension