/AI_final_project

Group work AI final project on Toxic Comment Classification dataset

Primary LanguageJupyter Notebook

AI_final_project

Group work AI final project on Toxic Comment Classification dataset

The project is to build a model capable of detecting the toxicity of a comment, assigning probabilities for different levels of toxicity. Our model is trained on a wikipedia dataset comprised of comments from Wikipedia's discussion pages.

You can acces our final presentation fllowing this link, or you can find the pdf on the github here.