├── .gitignore ├── LICENSE ├── README.md ├── asset └── model.png ├── data ├── dbpedia │ └── extract_kg.py ├── inspired │ ├── dev.jsonl │ ├── extract_subkg.py │ ├── relation_set.json │ ├── remove_entity.py │ ├── test.jsonl │ └── train.jsonl └── redial │ ├── extract_subkg.py │ ├── relation_set.json │ ├── remove_entity.py │ ├── test_data_dbpedia_raw.jsonl │ ├── train_data_dbpedia_raw.jsonl │ └── valid_data_dbpedia_raw.jsonl └── src ├── config.py ├── data ├── inspired │ ├── process.py │ └── process_mask.py ├── inspired_gen │ └── merge.py ├── redial │ ├── process.py │ └── process_mask.py └── redial_gen │ └── merge.py ├── dataset_conv.py ├── dataset_dbpedia.py ├── dataset_pre.py ├── dataset_rec.py ├── evaluate_conv.py ├── evaluate_rec.py ├── infer_conv.py ├── infer_conv.sh ├── model_gpt2.py ├── model_prompt.py ├── train_conv.py ├── train_conv.sh ├── train_pre.py ├── train_pre.sh ├── train_rec.py ├── train_rec.sh └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxl1999/UniCRS/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxl1999/UniCRS/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxl1999/UniCRS/HEAD/README.md -------------------------------------------------------------------------------- /asset/model.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxl1999/UniCRS/HEAD/asset/model.png -------------------------------------------------------------------------------- /data/dbpedia/extract_kg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxl1999/UniCRS/HEAD/data/dbpedia/extract_kg.py -------------------------------------------------------------------------------- /data/inspired/dev.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxl1999/UniCRS/HEAD/data/inspired/dev.jsonl -------------------------------------------------------------------------------- /data/inspired/extract_subkg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxl1999/UniCRS/HEAD/data/inspired/extract_subkg.py -------------------------------------------------------------------------------- /data/inspired/relation_set.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxl1999/UniCRS/HEAD/data/inspired/relation_set.json -------------------------------------------------------------------------------- /data/inspired/remove_entity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxl1999/UniCRS/HEAD/data/inspired/remove_entity.py -------------------------------------------------------------------------------- /data/inspired/test.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxl1999/UniCRS/HEAD/data/inspired/test.jsonl -------------------------------------------------------------------------------- /data/inspired/train.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxl1999/UniCRS/HEAD/data/inspired/train.jsonl -------------------------------------------------------------------------------- /data/redial/extract_subkg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxl1999/UniCRS/HEAD/data/redial/extract_subkg.py -------------------------------------------------------------------------------- /data/redial/relation_set.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxl1999/UniCRS/HEAD/data/redial/relation_set.json -------------------------------------------------------------------------------- /data/redial/remove_entity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxl1999/UniCRS/HEAD/data/redial/remove_entity.py -------------------------------------------------------------------------------- /data/redial/test_data_dbpedia_raw.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxl1999/UniCRS/HEAD/data/redial/test_data_dbpedia_raw.jsonl -------------------------------------------------------------------------------- /data/redial/train_data_dbpedia_raw.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxl1999/UniCRS/HEAD/data/redial/train_data_dbpedia_raw.jsonl -------------------------------------------------------------------------------- /data/redial/valid_data_dbpedia_raw.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxl1999/UniCRS/HEAD/data/redial/valid_data_dbpedia_raw.jsonl -------------------------------------------------------------------------------- /src/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxl1999/UniCRS/HEAD/src/config.py -------------------------------------------------------------------------------- /src/data/inspired/process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxl1999/UniCRS/HEAD/src/data/inspired/process.py -------------------------------------------------------------------------------- /src/data/inspired/process_mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxl1999/UniCRS/HEAD/src/data/inspired/process_mask.py -------------------------------------------------------------------------------- /src/data/inspired_gen/merge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxl1999/UniCRS/HEAD/src/data/inspired_gen/merge.py -------------------------------------------------------------------------------- /src/data/redial/process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxl1999/UniCRS/HEAD/src/data/redial/process.py -------------------------------------------------------------------------------- /src/data/redial/process_mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxl1999/UniCRS/HEAD/src/data/redial/process_mask.py -------------------------------------------------------------------------------- /src/data/redial_gen/merge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxl1999/UniCRS/HEAD/src/data/redial_gen/merge.py -------------------------------------------------------------------------------- /src/dataset_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxl1999/UniCRS/HEAD/src/dataset_conv.py -------------------------------------------------------------------------------- /src/dataset_dbpedia.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxl1999/UniCRS/HEAD/src/dataset_dbpedia.py -------------------------------------------------------------------------------- /src/dataset_pre.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxl1999/UniCRS/HEAD/src/dataset_pre.py -------------------------------------------------------------------------------- /src/dataset_rec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxl1999/UniCRS/HEAD/src/dataset_rec.py -------------------------------------------------------------------------------- /src/evaluate_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxl1999/UniCRS/HEAD/src/evaluate_conv.py -------------------------------------------------------------------------------- /src/evaluate_rec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxl1999/UniCRS/HEAD/src/evaluate_rec.py -------------------------------------------------------------------------------- /src/infer_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxl1999/UniCRS/HEAD/src/infer_conv.py -------------------------------------------------------------------------------- /src/infer_conv.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxl1999/UniCRS/HEAD/src/infer_conv.sh -------------------------------------------------------------------------------- /src/model_gpt2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxl1999/UniCRS/HEAD/src/model_gpt2.py -------------------------------------------------------------------------------- /src/model_prompt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxl1999/UniCRS/HEAD/src/model_prompt.py -------------------------------------------------------------------------------- /src/train_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxl1999/UniCRS/HEAD/src/train_conv.py -------------------------------------------------------------------------------- /src/train_conv.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxl1999/UniCRS/HEAD/src/train_conv.sh -------------------------------------------------------------------------------- /src/train_pre.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxl1999/UniCRS/HEAD/src/train_pre.py -------------------------------------------------------------------------------- /src/train_pre.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxl1999/UniCRS/HEAD/src/train_pre.sh -------------------------------------------------------------------------------- /src/train_rec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxl1999/UniCRS/HEAD/src/train_rec.py -------------------------------------------------------------------------------- /src/train_rec.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxl1999/UniCRS/HEAD/src/train_rec.sh -------------------------------------------------------------------------------- /src/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wxl1999/UniCRS/HEAD/src/utils.py --------------------------------------------------------------------------------