Skip to content

Code for our PAKDD'19 paper "Multi-task Learning for Target-dependent Sentiment Classification"

Notifications You must be signed in to change notification settings


Repository files navigation

Multi-task Learning for Target-dependent Sentiment Classification

Packaged datasets and Keras code for the paper Multi-task Learning for Target-dependent Sentiment Classification.

We use tensorflow-gpu-1.4.0 which needs cudnn6. To run on CUDA ca 2019, you need to download cudnn6 from here and install along with CUDA8.

Prepare a virtual environment and install requirements as follows.

$ virtualenv -p `which python2` /path/to/mttdsc_env
$ source /path/to/mttdsc_env/bin/activate
(mttdsc_env)$ pip install -r requirements.txt

We will assume this code has been cloned to /path/to/mttdsc as the code base directory. Download the zipped data files and unzip in the code base directory, which will place all the .h5 files in the data subdirectory. Gdrive can be used for downloading.

$ cd /tmp
$ gdrive download 18av-HZCx1G14CURRyYrjjmNcevOphaQc
$ cd /path/to/mttdsc
$ unzip /path/to/zipfile

If you want to prepare the data sets by yourself,

(mttdsc_env)$ cd /path/to/mttdsc/data_prep
(mttdsc_env)$ cd python

Training the models

To train the models use . Refer to the following commands

(mttdsc_env)$ python config_files/tdgru_lidong.json   # training TDGRU on Lidong Dataset
(mttdsc_env)$ python config_files/naive_mtl_lidong.json   # training NaiveMTL on Lidong Dataset
(mttdsc_env)$ python config_files/mttdsc_lidong.json   # training MTTDSC on Lidong Dataset

Test results would be printed towared the end of completion of the script.

UK Election dataset will be added shortly.

Using the pretrained model

We have provided an easy to use API to get the the sentiment of any sentence. The API automatically downloads the model weights. Refer to the following snippet to use our pretrained sentiment model.

from tdsc_pretrained import get_sentiment
print get_sentiment( 'i like taylor swift and her music is great'  , 'taylor swift'  )


Code for our PAKDD'19 paper "Multi-task Learning for Target-dependent Sentiment Classification"






No packages published