The technical report can be found here.
- Python >= 3.7
- torch >= 1.9.0
- numpy >= 1.17.2
- transformers >= 4.15.0
git clone https://github.com/hiyouga/Toxic_Detection.git
conda create -n toxic python=3.7
conda activate toxic
pip install -r requirements.txt
python data/split_data.py
python main.py
python main.py -h
This is a group homework for "Machine Learning" in BUAA Graduate School.
hiyouga [AT] buaa [DOT] edu [DOT] cn
MIT