This is the code for the paper:
VistaNet: Visual Aspect Attention Network for Multimodal Sentiment Analysis
Quoc-Tuan Truong and Hady W. Lauw
Presented at AAAI 2019
We provide:
- Code to train and evaluate the model
- Data used for the experiments
If you find the code and data useful in your research, please cite:
@inproceedings{truong2019vistanet,
title={Vistanet: Visual aspect attention network for multimodal sentiment analysis},
author={Truong, Quoc-Tuan and Lauw, Hady W},
booktitle={Proceedings of the AAAI conference on artificial intelligence},
volume={33},
number={01},
pages={305--312},
year={2019}
}
- Python 3
- Tensorflow >=1.12,<2.0
- Tqdm
- GloVe word embeddings
- Make sure the data is ready. Run script to pre-process the data:
python data_preprocess.py
- Train
VistaNet
:
python train.py --hidden_dim 50 --att_dim 100 --num_images 3 --batch_size 32 --learning_rate 0.001 --num_epochs 20
Questions and discussion are welcome: www.qttruong.com