├── KEFU3 ├── FactAwareWrapper.py ├── KEFUAttentionWrapperHGFU.py ├── KEFUWrapper.py ├── __init__.py ├── iterator_utils.py ├── iterator_utils2.py ├── kefu_classifier.py ├── kefu_classifier_rank.py ├── kefu_knowledge_utils.py ├── kefu_model3.py ├── knowledge_utils.py ├── run_classifier.py └── run_kefu3.py ├── README.md ├── config ├── Classifier.txt ├── Joint.txt ├── JointGumbel.txt ├── JointSample.txt ├── Round.txt ├── RoundGumbel.txt ├── RoundSample.txt ├── TwoStage.txt ├── train_classifier.sh ├── train_joint.sh ├── train_joint_gumbel.sh ├── train_joint_sample.sh ├── train_round.sh ├── train_round_gumbel.sh ├── train_round_sample.sh └── train_twostage.sh ├── eval.py ├── lib ├── NRGWrapper.py ├── __init__.py ├── config_parser.py ├── dataset_utils.py ├── eval.py ├── evaluation_scripts │ ├── __init__.py │ ├── bleu.py │ ├── embed.py │ ├── rank_evaluation.py │ ├── ref_bleu.py │ ├── rouge.py │ ├── statistical_eval.py │ └── tokens2wordlevel.py ├── evaluation_utils.py ├── layers │ ├── __init__.py │ └── beam_search.py ├── model_helper.py ├── status_counter.py ├── utils.py └── vocab_utils.py └── main.py /KEFU3/FactAwareWrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pku-sixing/IJCAI2020-TopicKA/HEAD/KEFU3/FactAwareWrapper.py -------------------------------------------------------------------------------- /KEFU3/KEFUAttentionWrapperHGFU.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pku-sixing/IJCAI2020-TopicKA/HEAD/KEFU3/KEFUAttentionWrapperHGFU.py -------------------------------------------------------------------------------- /KEFU3/KEFUWrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pku-sixing/IJCAI2020-TopicKA/HEAD/KEFU3/KEFUWrapper.py -------------------------------------------------------------------------------- /KEFU3/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /KEFU3/iterator_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pku-sixing/IJCAI2020-TopicKA/HEAD/KEFU3/iterator_utils.py -------------------------------------------------------------------------------- /KEFU3/iterator_utils2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pku-sixing/IJCAI2020-TopicKA/HEAD/KEFU3/iterator_utils2.py -------------------------------------------------------------------------------- /KEFU3/kefu_classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pku-sixing/IJCAI2020-TopicKA/HEAD/KEFU3/kefu_classifier.py -------------------------------------------------------------------------------- /KEFU3/kefu_classifier_rank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pku-sixing/IJCAI2020-TopicKA/HEAD/KEFU3/kefu_classifier_rank.py -------------------------------------------------------------------------------- /KEFU3/kefu_knowledge_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pku-sixing/IJCAI2020-TopicKA/HEAD/KEFU3/kefu_knowledge_utils.py -------------------------------------------------------------------------------- /KEFU3/kefu_model3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pku-sixing/IJCAI2020-TopicKA/HEAD/KEFU3/kefu_model3.py -------------------------------------------------------------------------------- /KEFU3/knowledge_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pku-sixing/IJCAI2020-TopicKA/HEAD/KEFU3/knowledge_utils.py -------------------------------------------------------------------------------- /KEFU3/run_classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pku-sixing/IJCAI2020-TopicKA/HEAD/KEFU3/run_classifier.py -------------------------------------------------------------------------------- /KEFU3/run_kefu3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pku-sixing/IJCAI2020-TopicKA/HEAD/KEFU3/run_kefu3.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pku-sixing/IJCAI2020-TopicKA/HEAD/README.md -------------------------------------------------------------------------------- /config/Classifier.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pku-sixing/IJCAI2020-TopicKA/HEAD/config/Classifier.txt -------------------------------------------------------------------------------- /config/Joint.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pku-sixing/IJCAI2020-TopicKA/HEAD/config/Joint.txt -------------------------------------------------------------------------------- /config/JointGumbel.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pku-sixing/IJCAI2020-TopicKA/HEAD/config/JointGumbel.txt -------------------------------------------------------------------------------- /config/JointSample.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pku-sixing/IJCAI2020-TopicKA/HEAD/config/JointSample.txt -------------------------------------------------------------------------------- /config/Round.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pku-sixing/IJCAI2020-TopicKA/HEAD/config/Round.txt -------------------------------------------------------------------------------- /config/RoundGumbel.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pku-sixing/IJCAI2020-TopicKA/HEAD/config/RoundGumbel.txt -------------------------------------------------------------------------------- /config/RoundSample.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pku-sixing/IJCAI2020-TopicKA/HEAD/config/RoundSample.txt -------------------------------------------------------------------------------- /config/TwoStage.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pku-sixing/IJCAI2020-TopicKA/HEAD/config/TwoStage.txt -------------------------------------------------------------------------------- /config/train_classifier.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pku-sixing/IJCAI2020-TopicKA/HEAD/config/train_classifier.sh -------------------------------------------------------------------------------- /config/train_joint.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pku-sixing/IJCAI2020-TopicKA/HEAD/config/train_joint.sh -------------------------------------------------------------------------------- /config/train_joint_gumbel.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pku-sixing/IJCAI2020-TopicKA/HEAD/config/train_joint_gumbel.sh -------------------------------------------------------------------------------- /config/train_joint_sample.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pku-sixing/IJCAI2020-TopicKA/HEAD/config/train_joint_sample.sh -------------------------------------------------------------------------------- /config/train_round.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pku-sixing/IJCAI2020-TopicKA/HEAD/config/train_round.sh -------------------------------------------------------------------------------- /config/train_round_gumbel.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pku-sixing/IJCAI2020-TopicKA/HEAD/config/train_round_gumbel.sh -------------------------------------------------------------------------------- /config/train_round_sample.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pku-sixing/IJCAI2020-TopicKA/HEAD/config/train_round_sample.sh -------------------------------------------------------------------------------- /config/train_twostage.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pku-sixing/IJCAI2020-TopicKA/HEAD/config/train_twostage.sh -------------------------------------------------------------------------------- /eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pku-sixing/IJCAI2020-TopicKA/HEAD/eval.py -------------------------------------------------------------------------------- /lib/NRGWrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pku-sixing/IJCAI2020-TopicKA/HEAD/lib/NRGWrapper.py -------------------------------------------------------------------------------- /lib/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/config_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pku-sixing/IJCAI2020-TopicKA/HEAD/lib/config_parser.py -------------------------------------------------------------------------------- /lib/dataset_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pku-sixing/IJCAI2020-TopicKA/HEAD/lib/dataset_utils.py -------------------------------------------------------------------------------- /lib/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pku-sixing/IJCAI2020-TopicKA/HEAD/lib/eval.py -------------------------------------------------------------------------------- /lib/evaluation_scripts/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/evaluation_scripts/bleu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pku-sixing/IJCAI2020-TopicKA/HEAD/lib/evaluation_scripts/bleu.py -------------------------------------------------------------------------------- /lib/evaluation_scripts/embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pku-sixing/IJCAI2020-TopicKA/HEAD/lib/evaluation_scripts/embed.py -------------------------------------------------------------------------------- /lib/evaluation_scripts/rank_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pku-sixing/IJCAI2020-TopicKA/HEAD/lib/evaluation_scripts/rank_evaluation.py -------------------------------------------------------------------------------- /lib/evaluation_scripts/ref_bleu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pku-sixing/IJCAI2020-TopicKA/HEAD/lib/evaluation_scripts/ref_bleu.py -------------------------------------------------------------------------------- /lib/evaluation_scripts/rouge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pku-sixing/IJCAI2020-TopicKA/HEAD/lib/evaluation_scripts/rouge.py -------------------------------------------------------------------------------- /lib/evaluation_scripts/statistical_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pku-sixing/IJCAI2020-TopicKA/HEAD/lib/evaluation_scripts/statistical_eval.py -------------------------------------------------------------------------------- /lib/evaluation_scripts/tokens2wordlevel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pku-sixing/IJCAI2020-TopicKA/HEAD/lib/evaluation_scripts/tokens2wordlevel.py -------------------------------------------------------------------------------- /lib/evaluation_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pku-sixing/IJCAI2020-TopicKA/HEAD/lib/evaluation_utils.py -------------------------------------------------------------------------------- /lib/layers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/layers/beam_search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pku-sixing/IJCAI2020-TopicKA/HEAD/lib/layers/beam_search.py -------------------------------------------------------------------------------- /lib/model_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pku-sixing/IJCAI2020-TopicKA/HEAD/lib/model_helper.py -------------------------------------------------------------------------------- /lib/status_counter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pku-sixing/IJCAI2020-TopicKA/HEAD/lib/status_counter.py -------------------------------------------------------------------------------- /lib/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pku-sixing/IJCAI2020-TopicKA/HEAD/lib/utils.py -------------------------------------------------------------------------------- /lib/vocab_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pku-sixing/IJCAI2020-TopicKA/HEAD/lib/vocab_utils.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pku-sixing/IJCAI2020-TopicKA/HEAD/main.py --------------------------------------------------------------------------------