An open API service indexing awesome lists of open source software.

https://github.com/ivanprodaiko94/ucu-machine-learning-course-final-project


https://github.com/ivanprodaiko94/ucu-machine-learning-course-final-project

Last synced: about 1 year ago
JSON representation

Awesome Lists containing this project

README

          

# 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.