/Toxic-Comment-Classification

CNN / LSTM / BERT

Primary LanguagePythonGNU General Public License v3.0GPL-3.0

Toxic-Comment-Classification

Requirements for training with GPU (in order of installation)

Steps

  • Clone the repository
  • Create a virtual environment
  • Install setuptools and wheel
  • Install requirements.txt
  • Download glove
  • Run create_custom_glove.py
  • Run main.py