├── .gitignore ├── LICENSE ├── README.md ├── data ├── KVR │ ├── .DS_Store │ ├── dev.txt │ ├── kvret_dev_public.json │ ├── kvret_entities.json │ ├── kvret_test_public.json │ ├── kvret_train_public.json │ ├── read_data.py │ ├── test.txt │ └── train.txt ├── dialog-bAbI-tasks │ ├── .DS_Store │ ├── LICENSE.txt │ ├── README.txt │ ├── dialog-babi-candidates.txt │ ├── dialog-babi-kb-all.txt │ ├── dialog-babi-task1dev.txt │ ├── dialog-babi-task1trn.txt │ ├── dialog-babi-task1tst-OOV.txt │ ├── dialog-babi-task1tst.txt │ ├── dialog-babi-task2dev.txt │ ├── dialog-babi-task2trn.txt │ ├── dialog-babi-task2tst-OOV.txt │ ├── dialog-babi-task2tst.txt │ ├── dialog-babi-task3dev.txt │ ├── dialog-babi-task3trn.txt │ ├── dialog-babi-task3tst-OOV.txt │ ├── dialog-babi-task3tst.txt │ ├── dialog-babi-task4dev.txt │ ├── dialog-babi-task4trn.txt │ ├── dialog-babi-task4tst-OOV.txt │ ├── dialog-babi-task4tst.txt │ ├── dialog-babi-task5dev.txt │ ├── dialog-babi-task5trn.txt │ ├── dialog-babi-task5tst-OOV.txt │ ├── dialog-babi-task5tst.txt │ ├── dialog-babi-task6-dstc2-candidates.txt │ ├── dialog-babi-task6-dstc2-kb.txt │ ├── dialog-babi-task6dev.txt │ ├── dialog-babi-task6trn.txt │ └── dialog-babi-task6tst.txt └── eng-fra.txt ├── img ├── Mem2seq.pdf ├── VIZ_MEMTINCAR.pdf ├── mem2seq.png └── viz.png ├── main_nmt.py ├── main_test.py ├── main_train.py ├── models ├── Mem2Seq.py ├── Mem2Seq_NMT.py ├── enc_Luong.py ├── enc_PTRUNK.py └── enc_vanilla.py ├── requirements.txt └── utils ├── config.py ├── masked_cross_entropy.py ├── measures.py ├── multi-bleu.perl ├── until_temp.py ├── utils_NMT.py ├── utils_babi.py ├── utils_babi_mem2seq.py ├── utils_kvr.py └── utils_kvr_mem2seq.py /.gitignore: -------------------------------------------------------------------------------- 1 | save/ 2 | *.pyc 3 | sftp-config.json 4 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/README.md -------------------------------------------------------------------------------- /data/KVR/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/data/KVR/.DS_Store -------------------------------------------------------------------------------- /data/KVR/dev.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/data/KVR/dev.txt -------------------------------------------------------------------------------- /data/KVR/kvret_dev_public.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/data/KVR/kvret_dev_public.json -------------------------------------------------------------------------------- /data/KVR/kvret_entities.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/data/KVR/kvret_entities.json -------------------------------------------------------------------------------- /data/KVR/kvret_test_public.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/data/KVR/kvret_test_public.json -------------------------------------------------------------------------------- /data/KVR/kvret_train_public.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/data/KVR/kvret_train_public.json -------------------------------------------------------------------------------- /data/KVR/read_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/data/KVR/read_data.py -------------------------------------------------------------------------------- /data/KVR/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/data/KVR/test.txt -------------------------------------------------------------------------------- /data/KVR/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/data/KVR/train.txt -------------------------------------------------------------------------------- /data/dialog-bAbI-tasks/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/data/dialog-bAbI-tasks/.DS_Store -------------------------------------------------------------------------------- /data/dialog-bAbI-tasks/LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/data/dialog-bAbI-tasks/LICENSE.txt -------------------------------------------------------------------------------- /data/dialog-bAbI-tasks/README.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/data/dialog-bAbI-tasks/README.txt -------------------------------------------------------------------------------- /data/dialog-bAbI-tasks/dialog-babi-candidates.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/data/dialog-bAbI-tasks/dialog-babi-candidates.txt -------------------------------------------------------------------------------- /data/dialog-bAbI-tasks/dialog-babi-kb-all.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/data/dialog-bAbI-tasks/dialog-babi-kb-all.txt -------------------------------------------------------------------------------- /data/dialog-bAbI-tasks/dialog-babi-task1dev.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/data/dialog-bAbI-tasks/dialog-babi-task1dev.txt -------------------------------------------------------------------------------- /data/dialog-bAbI-tasks/dialog-babi-task1trn.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/data/dialog-bAbI-tasks/dialog-babi-task1trn.txt -------------------------------------------------------------------------------- /data/dialog-bAbI-tasks/dialog-babi-task1tst-OOV.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/data/dialog-bAbI-tasks/dialog-babi-task1tst-OOV.txt -------------------------------------------------------------------------------- /data/dialog-bAbI-tasks/dialog-babi-task1tst.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/data/dialog-bAbI-tasks/dialog-babi-task1tst.txt -------------------------------------------------------------------------------- /data/dialog-bAbI-tasks/dialog-babi-task2dev.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/data/dialog-bAbI-tasks/dialog-babi-task2dev.txt -------------------------------------------------------------------------------- /data/dialog-bAbI-tasks/dialog-babi-task2trn.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/data/dialog-bAbI-tasks/dialog-babi-task2trn.txt -------------------------------------------------------------------------------- /data/dialog-bAbI-tasks/dialog-babi-task2tst-OOV.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/data/dialog-bAbI-tasks/dialog-babi-task2tst-OOV.txt -------------------------------------------------------------------------------- /data/dialog-bAbI-tasks/dialog-babi-task2tst.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/data/dialog-bAbI-tasks/dialog-babi-task2tst.txt -------------------------------------------------------------------------------- /data/dialog-bAbI-tasks/dialog-babi-task3dev.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/data/dialog-bAbI-tasks/dialog-babi-task3dev.txt -------------------------------------------------------------------------------- /data/dialog-bAbI-tasks/dialog-babi-task3trn.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/data/dialog-bAbI-tasks/dialog-babi-task3trn.txt -------------------------------------------------------------------------------- /data/dialog-bAbI-tasks/dialog-babi-task3tst-OOV.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/data/dialog-bAbI-tasks/dialog-babi-task3tst-OOV.txt -------------------------------------------------------------------------------- /data/dialog-bAbI-tasks/dialog-babi-task3tst.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/data/dialog-bAbI-tasks/dialog-babi-task3tst.txt -------------------------------------------------------------------------------- /data/dialog-bAbI-tasks/dialog-babi-task4dev.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/data/dialog-bAbI-tasks/dialog-babi-task4dev.txt -------------------------------------------------------------------------------- /data/dialog-bAbI-tasks/dialog-babi-task4trn.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/data/dialog-bAbI-tasks/dialog-babi-task4trn.txt -------------------------------------------------------------------------------- /data/dialog-bAbI-tasks/dialog-babi-task4tst-OOV.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/data/dialog-bAbI-tasks/dialog-babi-task4tst-OOV.txt -------------------------------------------------------------------------------- /data/dialog-bAbI-tasks/dialog-babi-task4tst.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/data/dialog-bAbI-tasks/dialog-babi-task4tst.txt -------------------------------------------------------------------------------- /data/dialog-bAbI-tasks/dialog-babi-task5dev.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/data/dialog-bAbI-tasks/dialog-babi-task5dev.txt -------------------------------------------------------------------------------- /data/dialog-bAbI-tasks/dialog-babi-task5trn.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/data/dialog-bAbI-tasks/dialog-babi-task5trn.txt -------------------------------------------------------------------------------- /data/dialog-bAbI-tasks/dialog-babi-task5tst-OOV.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/data/dialog-bAbI-tasks/dialog-babi-task5tst-OOV.txt -------------------------------------------------------------------------------- /data/dialog-bAbI-tasks/dialog-babi-task5tst.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/data/dialog-bAbI-tasks/dialog-babi-task5tst.txt -------------------------------------------------------------------------------- /data/dialog-bAbI-tasks/dialog-babi-task6-dstc2-candidates.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/data/dialog-bAbI-tasks/dialog-babi-task6-dstc2-candidates.txt -------------------------------------------------------------------------------- /data/dialog-bAbI-tasks/dialog-babi-task6-dstc2-kb.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/data/dialog-bAbI-tasks/dialog-babi-task6-dstc2-kb.txt -------------------------------------------------------------------------------- /data/dialog-bAbI-tasks/dialog-babi-task6dev.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/data/dialog-bAbI-tasks/dialog-babi-task6dev.txt -------------------------------------------------------------------------------- /data/dialog-bAbI-tasks/dialog-babi-task6trn.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/data/dialog-bAbI-tasks/dialog-babi-task6trn.txt -------------------------------------------------------------------------------- /data/dialog-bAbI-tasks/dialog-babi-task6tst.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/data/dialog-bAbI-tasks/dialog-babi-task6tst.txt -------------------------------------------------------------------------------- /data/eng-fra.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/data/eng-fra.txt -------------------------------------------------------------------------------- /img/Mem2seq.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/img/Mem2seq.pdf -------------------------------------------------------------------------------- /img/VIZ_MEMTINCAR.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/img/VIZ_MEMTINCAR.pdf -------------------------------------------------------------------------------- /img/mem2seq.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/img/mem2seq.png -------------------------------------------------------------------------------- /img/viz.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/img/viz.png -------------------------------------------------------------------------------- /main_nmt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/main_nmt.py -------------------------------------------------------------------------------- /main_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/main_test.py -------------------------------------------------------------------------------- /main_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/main_train.py -------------------------------------------------------------------------------- /models/Mem2Seq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/models/Mem2Seq.py -------------------------------------------------------------------------------- /models/Mem2Seq_NMT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/models/Mem2Seq_NMT.py -------------------------------------------------------------------------------- /models/enc_Luong.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/models/enc_Luong.py -------------------------------------------------------------------------------- /models/enc_PTRUNK.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/models/enc_PTRUNK.py -------------------------------------------------------------------------------- /models/enc_vanilla.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/models/enc_vanilla.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/requirements.txt -------------------------------------------------------------------------------- /utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/utils/config.py -------------------------------------------------------------------------------- /utils/masked_cross_entropy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/utils/masked_cross_entropy.py -------------------------------------------------------------------------------- /utils/measures.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/utils/measures.py -------------------------------------------------------------------------------- /utils/multi-bleu.perl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/utils/multi-bleu.perl -------------------------------------------------------------------------------- /utils/until_temp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/utils/until_temp.py -------------------------------------------------------------------------------- /utils/utils_NMT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/utils/utils_NMT.py -------------------------------------------------------------------------------- /utils/utils_babi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/utils/utils_babi.py -------------------------------------------------------------------------------- /utils/utils_babi_mem2seq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/utils/utils_babi_mem2seq.py -------------------------------------------------------------------------------- /utils/utils_kvr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/utils/utils_kvr.py -------------------------------------------------------------------------------- /utils/utils_kvr_mem2seq.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/HLTCHKUST/Mem2Seq/HEAD/utils/utils_kvr_mem2seq.py --------------------------------------------------------------------------------