/Hybrid-Code-Networks

Hybrid Code Networks in PyTorch

Primary LanguagePythonApache License 2.0Apache-2.0

Hybrid Code Networks (unofficial)

Hybrid Code Networks: practical and efficient end-to-end dialog control with supervised and reinforcement learning
Jason D. Williams, Kavosh Asadi, Geoffrey Zweig
https://arxiv.org/abs/1702.03274

This repo contains the implementation of Hybrid Code Networks in PyTorch.

Setup

You need to download following data.

./dialog-bAbI-tasks/dialog-babi-kb-all.txt
./dialog-bAbI-tasks/dialog-babi-task5-full-dialogs-trn.txt
./dialog-bAbI-tasks/dialog-babi-task5-full-dialogs-tst-OOV.txt

./data/GoogleNews-vectors-negative300.bin
  • Libraries
$ pip install tqdm

Train & Test

$ python train.py --help
$ python train.py
$ python train.py --test 1

TODOs

  • test Task6
  • Reinforcement Learning