Skip to content

Latest commit

 

History

History
executable file
·
36 lines (23 loc) · 1000 Bytes

README.md

File metadata and controls

executable file
·
36 lines (23 loc) · 1000 Bytes
Raw

GateMemN2N

Implementation of Gated End-to-End Memory Networks with sklearn-like interface using Tensorflow. Tasks are from the bAbl dataset.

Get Started

git clone https://github.com/cstghitpku/GateMemN2N.git

mkdir ./gmemn2n/data/
cd ./gmemn2n/data/
wget http://www.thespermwhale.com/jaseweston/babi/tasks_1-20_v1-2.tar.gz
tar xzvf ./tasks_1-20_v1-2.tar.gz

cd ../
python single.py

Examples

Running a single bAbI task

Running a joint model on all bAbI tasks

These files are also a good example of usage.

Requirements

  • tensorflow 1.0
  • scikit-learn 0.17.1
  • six 1.10.0

References

  • Julien Perez, Fei Liu, "Gated End-to-End Memory Networks", arXiv:1610.04211 [cs.CL].
  • Sainbayar Sukhbaatar, Arthur Szlam, Jason Weston, Rob Fergus, "End-To-End Memory Networks", arXiv:1503.08895 [cs.CL].