Toxic Comment Classification

主要功能

采用LSTM/C-LSTM/CNN等多种方法，对评论进行多标签分类

代码结构

main

• main.py
  无交叉验证，取训练误差最小的参数，用于测试

• main_fold.py
  k折交叉验证，测试时取k次的平均值

dataset

toxic comment

net

lstm/c-lstm/cnn...

运行

无交叉验证的运行指令如：

python main.py \
--server=6099 \
--phase=Train \
--sen_len=250 \
--net_name=lstm \
--dir_date=20180317 \
--lr_base=1e-3 \
--batch_size=256 \
--gpu=0 \

有交叉验证的指令如：

python main_fold.py \
--server=6099 \
--phase=Train \
--sen_len=250 \
--net_name=lstm \
--dir_date=20180318_fold_20 \
--batch_size=256 \
--lr_base=1e-3 \
--gpu=1 \
--fold_count=10 \