├── .gitignore ├── .nsmlignore ├── LICENSE.md ├── README.md ├── dataloader.py ├── dialog_output ├── index.html └── static │ ├── bootstrap.min.css │ ├── jquery-3.2.1.min.js │ └── main.js ├── env.yml ├── eval_utils ├── __init__.py ├── aqm_runner.py ├── dialog_generate.py ├── rank_answerer.py └── rank_questioner.py ├── evaluate.py ├── images ├── PMR-1.png ├── PMR-2.png ├── PMR-3.png ├── SL-PMR-1.png ├── dialog-1.png ├── dialog-2.png ├── dialog-3.png ├── dialog-4.png ├── dialog-5.png ├── iclr19-task-c.png ├── model_figure.jpg ├── qbot-match-gen.png ├── qbot-ours-gen.png ├── train-1.png ├── train-2.png ├── train-3.png └── visualization.jpg ├── options.py ├── pools.json ├── requirements.txt ├── scripts ├── download_checkpoints.sh └── download_preprocessed.sh ├── setup.py ├── train.json ├── train.py ├── utils ├── __init__.py ├── utilities.py └── visualize.py ├── val.json └── visdial ├── __init__.py ├── metrics.py └── models ├── __init__.py ├── agent.py ├── answerer.py ├── aqm_questioner.py ├── decoders ├── __init__.py └── gen.py ├── encoders ├── __init__.py ├── hre.py └── hre_real.py └── questioner.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver/aqm-plus/HEAD/.gitignore -------------------------------------------------------------------------------- /.nsmlignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver/aqm-plus/HEAD/.nsmlignore -------------------------------------------------------------------------------- /LICENSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver/aqm-plus/HEAD/LICENSE.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver/aqm-plus/HEAD/README.md -------------------------------------------------------------------------------- /dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver/aqm-plus/HEAD/dataloader.py -------------------------------------------------------------------------------- /dialog_output/index.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver/aqm-plus/HEAD/dialog_output/index.html -------------------------------------------------------------------------------- /dialog_output/static/bootstrap.min.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver/aqm-plus/HEAD/dialog_output/static/bootstrap.min.css -------------------------------------------------------------------------------- /dialog_output/static/jquery-3.2.1.min.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver/aqm-plus/HEAD/dialog_output/static/jquery-3.2.1.min.js -------------------------------------------------------------------------------- /dialog_output/static/main.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver/aqm-plus/HEAD/dialog_output/static/main.js -------------------------------------------------------------------------------- /env.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver/aqm-plus/HEAD/env.yml -------------------------------------------------------------------------------- /eval_utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /eval_utils/aqm_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver/aqm-plus/HEAD/eval_utils/aqm_runner.py -------------------------------------------------------------------------------- /eval_utils/dialog_generate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver/aqm-plus/HEAD/eval_utils/dialog_generate.py -------------------------------------------------------------------------------- /eval_utils/rank_answerer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver/aqm-plus/HEAD/eval_utils/rank_answerer.py -------------------------------------------------------------------------------- /eval_utils/rank_questioner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver/aqm-plus/HEAD/eval_utils/rank_questioner.py -------------------------------------------------------------------------------- /evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver/aqm-plus/HEAD/evaluate.py -------------------------------------------------------------------------------- /images/PMR-1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver/aqm-plus/HEAD/images/PMR-1.png -------------------------------------------------------------------------------- /images/PMR-2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver/aqm-plus/HEAD/images/PMR-2.png -------------------------------------------------------------------------------- /images/PMR-3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver/aqm-plus/HEAD/images/PMR-3.png -------------------------------------------------------------------------------- /images/SL-PMR-1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver/aqm-plus/HEAD/images/SL-PMR-1.png -------------------------------------------------------------------------------- /images/dialog-1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver/aqm-plus/HEAD/images/dialog-1.png -------------------------------------------------------------------------------- /images/dialog-2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver/aqm-plus/HEAD/images/dialog-2.png -------------------------------------------------------------------------------- /images/dialog-3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver/aqm-plus/HEAD/images/dialog-3.png -------------------------------------------------------------------------------- /images/dialog-4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver/aqm-plus/HEAD/images/dialog-4.png -------------------------------------------------------------------------------- /images/dialog-5.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver/aqm-plus/HEAD/images/dialog-5.png -------------------------------------------------------------------------------- /images/iclr19-task-c.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver/aqm-plus/HEAD/images/iclr19-task-c.png -------------------------------------------------------------------------------- /images/model_figure.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver/aqm-plus/HEAD/images/model_figure.jpg -------------------------------------------------------------------------------- /images/qbot-match-gen.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver/aqm-plus/HEAD/images/qbot-match-gen.png -------------------------------------------------------------------------------- /images/qbot-ours-gen.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver/aqm-plus/HEAD/images/qbot-ours-gen.png -------------------------------------------------------------------------------- /images/train-1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver/aqm-plus/HEAD/images/train-1.png -------------------------------------------------------------------------------- /images/train-2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver/aqm-plus/HEAD/images/train-2.png -------------------------------------------------------------------------------- /images/train-3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver/aqm-plus/HEAD/images/train-3.png -------------------------------------------------------------------------------- /images/visualization.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver/aqm-plus/HEAD/images/visualization.jpg -------------------------------------------------------------------------------- /options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver/aqm-plus/HEAD/options.py -------------------------------------------------------------------------------- /pools.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver/aqm-plus/HEAD/pools.json -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver/aqm-plus/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/download_checkpoints.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver/aqm-plus/HEAD/scripts/download_checkpoints.sh -------------------------------------------------------------------------------- /scripts/download_preprocessed.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver/aqm-plus/HEAD/scripts/download_preprocessed.sh -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver/aqm-plus/HEAD/setup.py -------------------------------------------------------------------------------- /train.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver/aqm-plus/HEAD/train.json -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver/aqm-plus/HEAD/train.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/utilities.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver/aqm-plus/HEAD/utils/utilities.py -------------------------------------------------------------------------------- /utils/visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver/aqm-plus/HEAD/utils/visualize.py -------------------------------------------------------------------------------- /val.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver/aqm-plus/HEAD/val.json -------------------------------------------------------------------------------- /visdial/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /visdial/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver/aqm-plus/HEAD/visdial/metrics.py -------------------------------------------------------------------------------- /visdial/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver/aqm-plus/HEAD/visdial/models/__init__.py -------------------------------------------------------------------------------- /visdial/models/agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver/aqm-plus/HEAD/visdial/models/agent.py -------------------------------------------------------------------------------- /visdial/models/answerer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver/aqm-plus/HEAD/visdial/models/answerer.py -------------------------------------------------------------------------------- /visdial/models/aqm_questioner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver/aqm-plus/HEAD/visdial/models/aqm_questioner.py -------------------------------------------------------------------------------- /visdial/models/decoders/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver/aqm-plus/HEAD/visdial/models/decoders/__init__.py -------------------------------------------------------------------------------- /visdial/models/decoders/gen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver/aqm-plus/HEAD/visdial/models/decoders/gen.py -------------------------------------------------------------------------------- /visdial/models/encoders/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver/aqm-plus/HEAD/visdial/models/encoders/__init__.py -------------------------------------------------------------------------------- /visdial/models/encoders/hre.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver/aqm-plus/HEAD/visdial/models/encoders/hre.py -------------------------------------------------------------------------------- /visdial/models/encoders/hre_real.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver/aqm-plus/HEAD/visdial/models/encoders/hre_real.py -------------------------------------------------------------------------------- /visdial/models/questioner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver/aqm-plus/HEAD/visdial/models/questioner.py --------------------------------------------------------------------------------