Skip to content

jacobbath/Hybrid-Code-Networks

 
 

Repository files navigation

Hybrid-Code-Networks

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

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%