/fudan_mtl_reviews

TensorFlow implementation of the paper `Adversarial Multi-task Learning for Text Classification`

Primary LanguagePython

Adversarial Multi-task Learning for Text Classification

TensorFlow implementation of the paper Adversarial Multi-task Learning for Text Classification.

performace using 50d word embedding:

models avg error
mtl 13.75
mtl + adv 12.79
mtl + adv + diff 12.70

To train the model:

cd data/
tar zxvf fudan-mtl-dataset.tar.gz
cd ../
python3 src/main.py --word_dim=50 --build_data
python3 src/main.py --word_dim=50 --adv
python3 src/main.py --word_dim=50 --adv --test