├── .gitignore ├── .gitmodules ├── LICENSE ├── README.md ├── parlai ├── README.md ├── __init__.py ├── agents │ ├── README.md │ ├── __init__.py │ ├── autorec │ │ ├── __init__.py │ │ ├── autorec.py │ │ └── modules.py │ ├── kbrd │ │ ├── __init__.py │ │ ├── kbrd.py │ │ └── modules.py │ ├── transformer │ │ ├── __init__.py │ │ ├── generator.py │ │ ├── modules.py │ │ ├── ranker.py │ │ └── transformer.py │ └── transformer_rec │ │ ├── __init__.py │ │ ├── generator.py │ │ ├── modules.py │ │ ├── ranker.py │ │ └── transformer_rec.py ├── core │ ├── README.md │ ├── __init__.py │ ├── agents.py │ ├── build_data.py │ ├── dict.py │ ├── distributed_utils.py │ ├── image_featurizers.py │ ├── logs.py │ ├── metrics.py │ ├── params.py │ ├── pytorch_data_teacher.py │ ├── teachers.py │ ├── testing_utils.py │ ├── thread_utils.py │ ├── torch_agent.py │ ├── torch_classifier_agent.py │ ├── torch_generator_agent.py │ ├── torch_ranker_agent.py │ ├── utils.py │ └── worlds.py ├── scripts │ ├── __init__.py │ ├── build_dict.py │ ├── build_pytorch_data.py │ ├── convert_data_to_fasttext_format.py │ ├── convert_data_to_parlai_format.py │ ├── data_stats.py │ ├── detect_offensive_language.py │ ├── display_data.py │ ├── display_model.py │ ├── distributed_train.py │ ├── eval_model.py │ ├── eval_ppl.py │ ├── eval_wordstat.py │ ├── extract_image_feature.py │ ├── interactive.py │ ├── interactive_rank.py │ ├── interactive_web.py │ ├── multiprocessing_train.py │ ├── profile_train.py │ ├── train_model.py │ └── verify_data.py └── tasks │ ├── README.md │ ├── __init__.py │ ├── redial │ ├── __init__.py │ ├── agents.py │ ├── build.py │ ├── extract_entities.py │ ├── gen_rippleset.py │ ├── train_autorec.py │ ├── train_autorec_tag.py │ ├── train_kbrd.py │ ├── train_rnn.py │ ├── train_transformer.py │ └── train_transformer_rec.py │ ├── task_list.py │ └── tasks.py ├── requirements.txt └── scripts ├── baseline.sh ├── both.sh ├── count.py ├── display_model.py ├── diversity_scorer.py ├── export_triples.py ├── score.py ├── show_bias.py ├── t2t_rec_eval.sh └── t2t_rec_rgcn.sh /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/.gitmodules -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/README.md -------------------------------------------------------------------------------- /parlai/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/README.md -------------------------------------------------------------------------------- /parlai/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/__init__.py -------------------------------------------------------------------------------- /parlai/agents/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/agents/README.md -------------------------------------------------------------------------------- /parlai/agents/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/agents/__init__.py -------------------------------------------------------------------------------- /parlai/agents/autorec/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /parlai/agents/autorec/autorec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/agents/autorec/autorec.py -------------------------------------------------------------------------------- /parlai/agents/autorec/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/agents/autorec/modules.py -------------------------------------------------------------------------------- /parlai/agents/kbrd/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /parlai/agents/kbrd/kbrd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/agents/kbrd/kbrd.py -------------------------------------------------------------------------------- /parlai/agents/kbrd/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/agents/kbrd/modules.py -------------------------------------------------------------------------------- /parlai/agents/transformer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/agents/transformer/__init__.py -------------------------------------------------------------------------------- /parlai/agents/transformer/generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/agents/transformer/generator.py -------------------------------------------------------------------------------- /parlai/agents/transformer/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/agents/transformer/modules.py -------------------------------------------------------------------------------- /parlai/agents/transformer/ranker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/agents/transformer/ranker.py -------------------------------------------------------------------------------- /parlai/agents/transformer/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/agents/transformer/transformer.py -------------------------------------------------------------------------------- /parlai/agents/transformer_rec/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/agents/transformer_rec/__init__.py -------------------------------------------------------------------------------- /parlai/agents/transformer_rec/generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/agents/transformer_rec/generator.py -------------------------------------------------------------------------------- /parlai/agents/transformer_rec/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/agents/transformer_rec/modules.py -------------------------------------------------------------------------------- /parlai/agents/transformer_rec/ranker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/agents/transformer_rec/ranker.py -------------------------------------------------------------------------------- /parlai/agents/transformer_rec/transformer_rec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/agents/transformer_rec/transformer_rec.py -------------------------------------------------------------------------------- /parlai/core/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/core/README.md -------------------------------------------------------------------------------- /parlai/core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/core/__init__.py -------------------------------------------------------------------------------- /parlai/core/agents.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/core/agents.py -------------------------------------------------------------------------------- /parlai/core/build_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/core/build_data.py -------------------------------------------------------------------------------- /parlai/core/dict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/core/dict.py -------------------------------------------------------------------------------- /parlai/core/distributed_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/core/distributed_utils.py -------------------------------------------------------------------------------- /parlai/core/image_featurizers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/core/image_featurizers.py -------------------------------------------------------------------------------- /parlai/core/logs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/core/logs.py -------------------------------------------------------------------------------- /parlai/core/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/core/metrics.py -------------------------------------------------------------------------------- /parlai/core/params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/core/params.py -------------------------------------------------------------------------------- /parlai/core/pytorch_data_teacher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/core/pytorch_data_teacher.py -------------------------------------------------------------------------------- /parlai/core/teachers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/core/teachers.py -------------------------------------------------------------------------------- /parlai/core/testing_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/core/testing_utils.py -------------------------------------------------------------------------------- /parlai/core/thread_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/core/thread_utils.py -------------------------------------------------------------------------------- /parlai/core/torch_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/core/torch_agent.py -------------------------------------------------------------------------------- /parlai/core/torch_classifier_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/core/torch_classifier_agent.py -------------------------------------------------------------------------------- /parlai/core/torch_generator_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/core/torch_generator_agent.py -------------------------------------------------------------------------------- /parlai/core/torch_ranker_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/core/torch_ranker_agent.py -------------------------------------------------------------------------------- /parlai/core/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/core/utils.py -------------------------------------------------------------------------------- /parlai/core/worlds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/core/worlds.py -------------------------------------------------------------------------------- /parlai/scripts/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/scripts/__init__.py -------------------------------------------------------------------------------- /parlai/scripts/build_dict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/scripts/build_dict.py -------------------------------------------------------------------------------- /parlai/scripts/build_pytorch_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/scripts/build_pytorch_data.py -------------------------------------------------------------------------------- /parlai/scripts/convert_data_to_fasttext_format.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/scripts/convert_data_to_fasttext_format.py -------------------------------------------------------------------------------- /parlai/scripts/convert_data_to_parlai_format.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/scripts/convert_data_to_parlai_format.py -------------------------------------------------------------------------------- /parlai/scripts/data_stats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/scripts/data_stats.py -------------------------------------------------------------------------------- /parlai/scripts/detect_offensive_language.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/scripts/detect_offensive_language.py -------------------------------------------------------------------------------- /parlai/scripts/display_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/scripts/display_data.py -------------------------------------------------------------------------------- /parlai/scripts/display_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/scripts/display_model.py -------------------------------------------------------------------------------- /parlai/scripts/distributed_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/scripts/distributed_train.py -------------------------------------------------------------------------------- /parlai/scripts/eval_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/scripts/eval_model.py -------------------------------------------------------------------------------- /parlai/scripts/eval_ppl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/scripts/eval_ppl.py -------------------------------------------------------------------------------- /parlai/scripts/eval_wordstat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/scripts/eval_wordstat.py -------------------------------------------------------------------------------- /parlai/scripts/extract_image_feature.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/scripts/extract_image_feature.py -------------------------------------------------------------------------------- /parlai/scripts/interactive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/scripts/interactive.py -------------------------------------------------------------------------------- /parlai/scripts/interactive_rank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/scripts/interactive_rank.py -------------------------------------------------------------------------------- /parlai/scripts/interactive_web.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/scripts/interactive_web.py -------------------------------------------------------------------------------- /parlai/scripts/multiprocessing_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/scripts/multiprocessing_train.py -------------------------------------------------------------------------------- /parlai/scripts/profile_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/scripts/profile_train.py -------------------------------------------------------------------------------- /parlai/scripts/train_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/scripts/train_model.py -------------------------------------------------------------------------------- /parlai/scripts/verify_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/scripts/verify_data.py -------------------------------------------------------------------------------- /parlai/tasks/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/tasks/README.md -------------------------------------------------------------------------------- /parlai/tasks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/tasks/__init__.py -------------------------------------------------------------------------------- /parlai/tasks/redial/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /parlai/tasks/redial/agents.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/tasks/redial/agents.py -------------------------------------------------------------------------------- /parlai/tasks/redial/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/tasks/redial/build.py -------------------------------------------------------------------------------- /parlai/tasks/redial/extract_entities.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/tasks/redial/extract_entities.py -------------------------------------------------------------------------------- /parlai/tasks/redial/gen_rippleset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/tasks/redial/gen_rippleset.py -------------------------------------------------------------------------------- /parlai/tasks/redial/train_autorec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/tasks/redial/train_autorec.py -------------------------------------------------------------------------------- /parlai/tasks/redial/train_autorec_tag.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/tasks/redial/train_autorec_tag.py -------------------------------------------------------------------------------- /parlai/tasks/redial/train_kbrd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/tasks/redial/train_kbrd.py -------------------------------------------------------------------------------- /parlai/tasks/redial/train_rnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/tasks/redial/train_rnn.py -------------------------------------------------------------------------------- /parlai/tasks/redial/train_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/tasks/redial/train_transformer.py -------------------------------------------------------------------------------- /parlai/tasks/redial/train_transformer_rec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/tasks/redial/train_transformer_rec.py -------------------------------------------------------------------------------- /parlai/tasks/task_list.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/tasks/task_list.py -------------------------------------------------------------------------------- /parlai/tasks/tasks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/parlai/tasks/tasks.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/baseline.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/scripts/baseline.sh -------------------------------------------------------------------------------- /scripts/both.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/scripts/both.sh -------------------------------------------------------------------------------- /scripts/count.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/scripts/count.py -------------------------------------------------------------------------------- /scripts/display_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/scripts/display_model.py -------------------------------------------------------------------------------- /scripts/diversity_scorer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/scripts/diversity_scorer.py -------------------------------------------------------------------------------- /scripts/export_triples.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/scripts/export_triples.py -------------------------------------------------------------------------------- /scripts/score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/scripts/score.py -------------------------------------------------------------------------------- /scripts/show_bias.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/scripts/show_bias.py -------------------------------------------------------------------------------- /scripts/t2t_rec_eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/scripts/t2t_rec_eval.sh -------------------------------------------------------------------------------- /scripts/t2t_rec_rgcn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUDM/KBRD/HEAD/scripts/t2t_rec_rgcn.sh --------------------------------------------------------------------------------