├── README.md ├── __init__.py ├── config.py ├── data └── kp20k_small │ ├── test_src.txt │ ├── test_trg.txt │ ├── train_src.txt │ ├── train_trg.txt │ ├── valid_src.txt │ └── valid_trg.txt ├── evaluate_prediction.py ├── images └── model.png ├── inference ├── __init__.py ├── beam.py ├── evaluate.py ├── penalties.py ├── sequence_generator.py └── set_generator.py ├── predict.py ├── preprocess.py ├── pykp ├── __init__.py ├── decoder │ ├── __init__.py │ └── transformer.py ├── encoder │ ├── __init__.py │ └── transformer.py ├── model.py ├── modules │ ├── __init__.py │ ├── multi_head_attn.py │ └── position_embed.py └── utils │ ├── __init__.py │ ├── io.py │ ├── label_assign.py │ ├── masked_loss.py │ └── seq2seq_state.py ├── scripts ├── run_one2seq.sh ├── run_one2set.sh ├── run_small_one2seq.sh └── run_small_one2set.sh ├── train.py ├── train_ml.py └── utils ├── __init__.py ├── data_loader.py ├── functions.py ├── report.py ├── statistics.py └── string_helper.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiacheng-ye/kg_one2set/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiacheng-ye/kg_one2set/HEAD/config.py -------------------------------------------------------------------------------- /data/kp20k_small/test_src.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiacheng-ye/kg_one2set/HEAD/data/kp20k_small/test_src.txt -------------------------------------------------------------------------------- /data/kp20k_small/test_trg.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiacheng-ye/kg_one2set/HEAD/data/kp20k_small/test_trg.txt -------------------------------------------------------------------------------- /data/kp20k_small/train_src.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiacheng-ye/kg_one2set/HEAD/data/kp20k_small/train_src.txt -------------------------------------------------------------------------------- /data/kp20k_small/train_trg.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiacheng-ye/kg_one2set/HEAD/data/kp20k_small/train_trg.txt -------------------------------------------------------------------------------- /data/kp20k_small/valid_src.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiacheng-ye/kg_one2set/HEAD/data/kp20k_small/valid_src.txt -------------------------------------------------------------------------------- /data/kp20k_small/valid_trg.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiacheng-ye/kg_one2set/HEAD/data/kp20k_small/valid_trg.txt -------------------------------------------------------------------------------- /evaluate_prediction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiacheng-ye/kg_one2set/HEAD/evaluate_prediction.py -------------------------------------------------------------------------------- /images/model.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiacheng-ye/kg_one2set/HEAD/images/model.png -------------------------------------------------------------------------------- /inference/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /inference/beam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiacheng-ye/kg_one2set/HEAD/inference/beam.py -------------------------------------------------------------------------------- /inference/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiacheng-ye/kg_one2set/HEAD/inference/evaluate.py -------------------------------------------------------------------------------- /inference/penalties.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiacheng-ye/kg_one2set/HEAD/inference/penalties.py -------------------------------------------------------------------------------- /inference/sequence_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiacheng-ye/kg_one2set/HEAD/inference/sequence_generator.py -------------------------------------------------------------------------------- /inference/set_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiacheng-ye/kg_one2set/HEAD/inference/set_generator.py -------------------------------------------------------------------------------- /predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiacheng-ye/kg_one2set/HEAD/predict.py -------------------------------------------------------------------------------- /preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiacheng-ye/kg_one2set/HEAD/preprocess.py -------------------------------------------------------------------------------- /pykp/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pykp/decoder/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pykp/decoder/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiacheng-ye/kg_one2set/HEAD/pykp/decoder/transformer.py -------------------------------------------------------------------------------- /pykp/encoder/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pykp/encoder/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiacheng-ye/kg_one2set/HEAD/pykp/encoder/transformer.py -------------------------------------------------------------------------------- /pykp/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiacheng-ye/kg_one2set/HEAD/pykp/model.py -------------------------------------------------------------------------------- /pykp/modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pykp/modules/multi_head_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiacheng-ye/kg_one2set/HEAD/pykp/modules/multi_head_attn.py -------------------------------------------------------------------------------- /pykp/modules/position_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiacheng-ye/kg_one2set/HEAD/pykp/modules/position_embed.py -------------------------------------------------------------------------------- /pykp/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pykp/utils/io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiacheng-ye/kg_one2set/HEAD/pykp/utils/io.py -------------------------------------------------------------------------------- /pykp/utils/label_assign.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiacheng-ye/kg_one2set/HEAD/pykp/utils/label_assign.py -------------------------------------------------------------------------------- /pykp/utils/masked_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiacheng-ye/kg_one2set/HEAD/pykp/utils/masked_loss.py -------------------------------------------------------------------------------- /pykp/utils/seq2seq_state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiacheng-ye/kg_one2set/HEAD/pykp/utils/seq2seq_state.py -------------------------------------------------------------------------------- /scripts/run_one2seq.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiacheng-ye/kg_one2set/HEAD/scripts/run_one2seq.sh -------------------------------------------------------------------------------- /scripts/run_one2set.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiacheng-ye/kg_one2set/HEAD/scripts/run_one2set.sh -------------------------------------------------------------------------------- /scripts/run_small_one2seq.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiacheng-ye/kg_one2set/HEAD/scripts/run_small_one2seq.sh -------------------------------------------------------------------------------- /scripts/run_small_one2set.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiacheng-ye/kg_one2set/HEAD/scripts/run_small_one2set.sh -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiacheng-ye/kg_one2set/HEAD/train.py -------------------------------------------------------------------------------- /train_ml.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiacheng-ye/kg_one2set/HEAD/train_ml.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiacheng-ye/kg_one2set/HEAD/utils/data_loader.py -------------------------------------------------------------------------------- /utils/functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiacheng-ye/kg_one2set/HEAD/utils/functions.py -------------------------------------------------------------------------------- /utils/report.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiacheng-ye/kg_one2set/HEAD/utils/report.py -------------------------------------------------------------------------------- /utils/statistics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiacheng-ye/kg_one2set/HEAD/utils/statistics.py -------------------------------------------------------------------------------- /utils/string_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiacheng-ye/kg_one2set/HEAD/utils/string_helper.py --------------------------------------------------------------------------------