GPU-accelerated PyTorch implementation of Zero-shot User Intent Detection via Capsule Neural Networks
This repository implements a capsule model IntentCapsNet-ZSL on the SNIPS-NLU dataset in Python 3 with PyTorch, first introduced in the paper Zero-shot User Intent Detection via Capsule Neural Networks.
The code aims to follow PyTorch best practices, using torch
instead of numpy
where possible, and using .cuda()
for GPU computation. Feel free to contribute via pull requests.
Congying Xia, Chenwei Zhang, Xiaohui Yan, Yi Chang, Philip S. Yu. Zero-shot User Intent Detection via Capsule Neural Networks. In Proceedings of the 2018 Conference on Empirical Methods in Natural Language Processing (EMNLP), 2018.
@article{xia2018zero,
title={Zero-shot User Intent Detection via Capsule Neural Networks},
author={Xia, Congying