├── LICENSE ├── README.md ├── imgs ├── model.jpg └── only_last_i3.pdf ├── main_re.py ├── models ├── __init__.py ├── __init__.pyc ├── __pycache__ │ ├── __init__.cpython-37.pyc │ ├── __init__.cpython-38.pyc │ ├── criterions.cpython-37.pyc │ ├── criterions.cpython-38.pyc │ ├── dialog_utils.cpython-37.pyc │ ├── dialog_utils.cpython-38.pyc │ ├── evaluators.cpython-37.pyc │ ├── evaluators.cpython-38.pyc │ ├── gpt.cpython-37.pyc │ ├── gpt.cpython-38.pyc │ ├── main.cpython-37.pyc │ ├── main.cpython-38.pyc │ ├── main_aggresive.cpython-37.pyc │ ├── main_aggresive.cpython-38.pyc │ ├── main_fs.cpython-37.pyc │ ├── main_fs.cpython-38.pyc │ ├── main_pretrain.cpython-37.pyc │ ├── main_pretrain.cpython-38.pyc │ ├── multi_bleu.cpython-37.pyc │ ├── multi_bleu.cpython-38.pyc │ ├── nn_lib.cpython-37.pyc │ ├── nn_lib.cpython-38.pyc │ ├── options.cpython-37.pyc │ ├── options.cpython-38.pyc │ ├── utils.cpython-37.pyc │ ├── utils.cpython-38.pyc │ ├── utt_utils.cpython-37.pyc │ └── utt_utils.cpython-38.pyc ├── criterions.py ├── dataset │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-37.pyc │ │ ├── __init__.cpython-38.pyc │ │ ├── corpora.cpython-37.pyc │ │ ├── corpora.cpython-38.pyc │ │ ├── data_loaders.cpython-37.pyc │ │ ├── data_loaders.cpython-38.pyc │ │ ├── dataloader_bases.cpython-37.pyc │ │ └── dataloader_bases.cpython-38.pyc │ ├── corpora.py │ ├── data_filter.py │ ├── data_loaders.py │ └── dataloader_bases.py ├── dialog_models.py ├── dialog_utils.py ├── enc2dec │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-37.pyc │ │ ├── __init__.cpython-38.pyc │ │ ├── base_modules.cpython-37.pyc │ │ ├── base_modules.cpython-38.pyc │ │ ├── decoders.cpython-37.pyc │ │ ├── decoders.cpython-38.pyc │ │ ├── encoders.cpython-37.pyc │ │ └── encoders.cpython-38.pyc │ ├── base_modules.py │ ├── decoders.py │ └── encoders.py ├── evaluators.py ├── evaluators.pyc ├── gpt.py ├── main.py ├── main_aggresive.py ├── main_fs.py ├── main_pretrain.py ├── main_semi.py ├── multi_bleu.py ├── nn_lib.py ├── options.py ├── utils.py ├── utt_skip_utils.py └── utt_utils.py └── requirements.txt /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/README.md -------------------------------------------------------------------------------- /imgs/model.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/imgs/model.jpg -------------------------------------------------------------------------------- /imgs/only_last_i3.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/imgs/only_last_i3.pdf -------------------------------------------------------------------------------- /main_re.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/main_re.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /models/__init__.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/__init__.pyc -------------------------------------------------------------------------------- /models/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /models/__pycache__/criterions.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/__pycache__/criterions.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/criterions.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/__pycache__/criterions.cpython-38.pyc -------------------------------------------------------------------------------- /models/__pycache__/dialog_utils.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/__pycache__/dialog_utils.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/dialog_utils.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/__pycache__/dialog_utils.cpython-38.pyc -------------------------------------------------------------------------------- /models/__pycache__/evaluators.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/__pycache__/evaluators.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/evaluators.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/__pycache__/evaluators.cpython-38.pyc -------------------------------------------------------------------------------- /models/__pycache__/gpt.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/__pycache__/gpt.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/gpt.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/__pycache__/gpt.cpython-38.pyc -------------------------------------------------------------------------------- /models/__pycache__/main.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/__pycache__/main.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/main.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/__pycache__/main.cpython-38.pyc -------------------------------------------------------------------------------- /models/__pycache__/main_aggresive.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/__pycache__/main_aggresive.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/main_aggresive.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/__pycache__/main_aggresive.cpython-38.pyc -------------------------------------------------------------------------------- /models/__pycache__/main_fs.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/__pycache__/main_fs.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/main_fs.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/__pycache__/main_fs.cpython-38.pyc -------------------------------------------------------------------------------- /models/__pycache__/main_pretrain.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/__pycache__/main_pretrain.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/main_pretrain.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/__pycache__/main_pretrain.cpython-38.pyc -------------------------------------------------------------------------------- /models/__pycache__/multi_bleu.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/__pycache__/multi_bleu.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/multi_bleu.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/__pycache__/multi_bleu.cpython-38.pyc -------------------------------------------------------------------------------- /models/__pycache__/nn_lib.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/__pycache__/nn_lib.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/nn_lib.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/__pycache__/nn_lib.cpython-38.pyc -------------------------------------------------------------------------------- /models/__pycache__/options.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/__pycache__/options.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/options.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/__pycache__/options.cpython-38.pyc -------------------------------------------------------------------------------- /models/__pycache__/utils.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/__pycache__/utils.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/utils.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/__pycache__/utils.cpython-38.pyc -------------------------------------------------------------------------------- /models/__pycache__/utt_utils.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/__pycache__/utt_utils.cpython-37.pyc -------------------------------------------------------------------------------- /models/__pycache__/utt_utils.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/__pycache__/utt_utils.cpython-38.pyc -------------------------------------------------------------------------------- /models/criterions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/criterions.py -------------------------------------------------------------------------------- /models/dataset/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/dataset/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/dataset/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /models/dataset/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/dataset/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /models/dataset/__pycache__/corpora.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/dataset/__pycache__/corpora.cpython-37.pyc -------------------------------------------------------------------------------- /models/dataset/__pycache__/corpora.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/dataset/__pycache__/corpora.cpython-38.pyc -------------------------------------------------------------------------------- /models/dataset/__pycache__/data_loaders.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/dataset/__pycache__/data_loaders.cpython-37.pyc -------------------------------------------------------------------------------- /models/dataset/__pycache__/data_loaders.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/dataset/__pycache__/data_loaders.cpython-38.pyc -------------------------------------------------------------------------------- /models/dataset/__pycache__/dataloader_bases.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/dataset/__pycache__/dataloader_bases.cpython-37.pyc -------------------------------------------------------------------------------- /models/dataset/__pycache__/dataloader_bases.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/dataset/__pycache__/dataloader_bases.cpython-38.pyc -------------------------------------------------------------------------------- /models/dataset/corpora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/dataset/corpora.py -------------------------------------------------------------------------------- /models/dataset/data_filter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/dataset/data_filter.py -------------------------------------------------------------------------------- /models/dataset/data_loaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/dataset/data_loaders.py -------------------------------------------------------------------------------- /models/dataset/dataloader_bases.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/dataset/dataloader_bases.py -------------------------------------------------------------------------------- /models/dialog_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/dialog_models.py -------------------------------------------------------------------------------- /models/dialog_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/dialog_utils.py -------------------------------------------------------------------------------- /models/enc2dec/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/enc2dec/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/enc2dec/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /models/enc2dec/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/enc2dec/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /models/enc2dec/__pycache__/base_modules.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/enc2dec/__pycache__/base_modules.cpython-37.pyc -------------------------------------------------------------------------------- /models/enc2dec/__pycache__/base_modules.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/enc2dec/__pycache__/base_modules.cpython-38.pyc -------------------------------------------------------------------------------- /models/enc2dec/__pycache__/decoders.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/enc2dec/__pycache__/decoders.cpython-37.pyc -------------------------------------------------------------------------------- /models/enc2dec/__pycache__/decoders.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/enc2dec/__pycache__/decoders.cpython-38.pyc -------------------------------------------------------------------------------- /models/enc2dec/__pycache__/encoders.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/enc2dec/__pycache__/encoders.cpython-37.pyc -------------------------------------------------------------------------------- /models/enc2dec/__pycache__/encoders.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/enc2dec/__pycache__/encoders.cpython-38.pyc -------------------------------------------------------------------------------- /models/enc2dec/base_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/enc2dec/base_modules.py -------------------------------------------------------------------------------- /models/enc2dec/decoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/enc2dec/decoders.py -------------------------------------------------------------------------------- /models/enc2dec/encoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/enc2dec/encoders.py -------------------------------------------------------------------------------- /models/evaluators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/evaluators.py -------------------------------------------------------------------------------- /models/evaluators.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/evaluators.pyc -------------------------------------------------------------------------------- /models/gpt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/gpt.py -------------------------------------------------------------------------------- /models/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/main.py -------------------------------------------------------------------------------- /models/main_aggresive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/main_aggresive.py -------------------------------------------------------------------------------- /models/main_fs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/main_fs.py -------------------------------------------------------------------------------- /models/main_pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/main_pretrain.py -------------------------------------------------------------------------------- /models/main_semi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/main_semi.py -------------------------------------------------------------------------------- /models/multi_bleu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/multi_bleu.py -------------------------------------------------------------------------------- /models/nn_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/nn_lib.py -------------------------------------------------------------------------------- /models/options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/options.py -------------------------------------------------------------------------------- /models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/utils.py -------------------------------------------------------------------------------- /models/utt_skip_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/utt_skip_utils.py -------------------------------------------------------------------------------- /models/utt_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/models/utt_utils.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/oklen/PEVAE/HEAD/requirements.txt --------------------------------------------------------------------------------