Skip to content

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

Notifications You must be signed in to change notification settings

FrankWork/fudan_mtl_reviews

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

About

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

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages