/UCU-machine-learning-course-final-project

Primary LanguageJupyter Notebook

Toxic comment classification

Collaborations:

  • Anton Bilchuk
  • Ivan Prodaiko

Run

ATTENTION: for running CUDA supported device is required!

To run this project you need test, train data and also pretrained embedding matrix. For this we recomend downlaod all needed data byt link https://drive.google.com/drive/folders/1QRKXty1zHTFeZIZIupbl43_ef36SiHtI?usp=sharing

Test model

For testing mdoel you can either run all train step and use this model or by link above use pretrained weights like in test_model.ipynb file.