├── LICENSE ├── MindSpore ├── eval_script_msqa.py ├── models │ ├── modeling_t5.py │ ├── modeling_t5_branching.py │ ├── modeling_utils.py │ └── utils.py ├── read_datasets.py ├── run_one2branch_qa.py ├── run_one2seq_qa.py └── utils.py ├── README.md ├── data ├── in_house │ └── msqa │ │ ├── dev.json │ │ ├── test.json │ │ └── train.json ├── kp20k_small │ ├── dev.json │ ├── test.json │ ├── train.json │ └── train_pred_large6.json ├── kptimes_small │ ├── dev.json │ ├── test.json │ └── train.json └── stackex_small │ ├── dev.json │ ├── test.json │ └── train.json ├── decoding.png ├── eval_scripts ├── eval_script_keyphrase.py ├── eval_script_msqa.py ├── kp_config.py ├── kp_utils │ └── string_helper.py └── pykp │ ├── __init__.py │ ├── model.py │ └── utils │ └── io.py ├── models ├── generation │ ├── configuration_utils.py │ └── utils.py ├── modeling_t5_branching.py └── modeling_utils.py ├── read_datasets.py ├── requirements.txt ├── run_one2branch_kp.py ├── run_one2branch_qa.py ├── run_one2seq_qa.py ├── script_branch_kp.sh ├── script_branch_qa.sh └── utils.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nju-websoft/One2Branch/HEAD/LICENSE -------------------------------------------------------------------------------- /MindSpore/eval_script_msqa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nju-websoft/One2Branch/HEAD/MindSpore/eval_script_msqa.py -------------------------------------------------------------------------------- /MindSpore/models/modeling_t5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nju-websoft/One2Branch/HEAD/MindSpore/models/modeling_t5.py -------------------------------------------------------------------------------- /MindSpore/models/modeling_t5_branching.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nju-websoft/One2Branch/HEAD/MindSpore/models/modeling_t5_branching.py -------------------------------------------------------------------------------- /MindSpore/models/modeling_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nju-websoft/One2Branch/HEAD/MindSpore/models/modeling_utils.py -------------------------------------------------------------------------------- /MindSpore/models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nju-websoft/One2Branch/HEAD/MindSpore/models/utils.py -------------------------------------------------------------------------------- /MindSpore/read_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nju-websoft/One2Branch/HEAD/MindSpore/read_datasets.py -------------------------------------------------------------------------------- /MindSpore/run_one2branch_qa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nju-websoft/One2Branch/HEAD/MindSpore/run_one2branch_qa.py -------------------------------------------------------------------------------- /MindSpore/run_one2seq_qa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nju-websoft/One2Branch/HEAD/MindSpore/run_one2seq_qa.py -------------------------------------------------------------------------------- /MindSpore/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nju-websoft/One2Branch/HEAD/MindSpore/utils.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nju-websoft/One2Branch/HEAD/README.md -------------------------------------------------------------------------------- /data/in_house/msqa/dev.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nju-websoft/One2Branch/HEAD/data/in_house/msqa/dev.json -------------------------------------------------------------------------------- /data/in_house/msqa/test.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nju-websoft/One2Branch/HEAD/data/in_house/msqa/test.json -------------------------------------------------------------------------------- /data/in_house/msqa/train.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nju-websoft/One2Branch/HEAD/data/in_house/msqa/train.json -------------------------------------------------------------------------------- /data/kp20k_small/dev.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nju-websoft/One2Branch/HEAD/data/kp20k_small/dev.json -------------------------------------------------------------------------------- /data/kp20k_small/test.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nju-websoft/One2Branch/HEAD/data/kp20k_small/test.json -------------------------------------------------------------------------------- /data/kp20k_small/train.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nju-websoft/One2Branch/HEAD/data/kp20k_small/train.json -------------------------------------------------------------------------------- /data/kp20k_small/train_pred_large6.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nju-websoft/One2Branch/HEAD/data/kp20k_small/train_pred_large6.json -------------------------------------------------------------------------------- /data/kptimes_small/dev.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nju-websoft/One2Branch/HEAD/data/kptimes_small/dev.json -------------------------------------------------------------------------------- /data/kptimes_small/test.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nju-websoft/One2Branch/HEAD/data/kptimes_small/test.json -------------------------------------------------------------------------------- /data/kptimes_small/train.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nju-websoft/One2Branch/HEAD/data/kptimes_small/train.json -------------------------------------------------------------------------------- /data/stackex_small/dev.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nju-websoft/One2Branch/HEAD/data/stackex_small/dev.json -------------------------------------------------------------------------------- /data/stackex_small/test.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nju-websoft/One2Branch/HEAD/data/stackex_small/test.json -------------------------------------------------------------------------------- /data/stackex_small/train.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nju-websoft/One2Branch/HEAD/data/stackex_small/train.json -------------------------------------------------------------------------------- /decoding.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nju-websoft/One2Branch/HEAD/decoding.png -------------------------------------------------------------------------------- /eval_scripts/eval_script_keyphrase.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nju-websoft/One2Branch/HEAD/eval_scripts/eval_script_keyphrase.py -------------------------------------------------------------------------------- /eval_scripts/eval_script_msqa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nju-websoft/One2Branch/HEAD/eval_scripts/eval_script_msqa.py -------------------------------------------------------------------------------- /eval_scripts/kp_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nju-websoft/One2Branch/HEAD/eval_scripts/kp_config.py -------------------------------------------------------------------------------- /eval_scripts/kp_utils/string_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nju-websoft/One2Branch/HEAD/eval_scripts/kp_utils/string_helper.py -------------------------------------------------------------------------------- /eval_scripts/pykp/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /eval_scripts/pykp/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nju-websoft/One2Branch/HEAD/eval_scripts/pykp/model.py -------------------------------------------------------------------------------- /eval_scripts/pykp/utils/io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nju-websoft/One2Branch/HEAD/eval_scripts/pykp/utils/io.py -------------------------------------------------------------------------------- /models/generation/configuration_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nju-websoft/One2Branch/HEAD/models/generation/configuration_utils.py -------------------------------------------------------------------------------- /models/generation/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nju-websoft/One2Branch/HEAD/models/generation/utils.py -------------------------------------------------------------------------------- /models/modeling_t5_branching.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nju-websoft/One2Branch/HEAD/models/modeling_t5_branching.py -------------------------------------------------------------------------------- /models/modeling_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nju-websoft/One2Branch/HEAD/models/modeling_utils.py -------------------------------------------------------------------------------- /read_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nju-websoft/One2Branch/HEAD/read_datasets.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nju-websoft/One2Branch/HEAD/requirements.txt -------------------------------------------------------------------------------- /run_one2branch_kp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nju-websoft/One2Branch/HEAD/run_one2branch_kp.py -------------------------------------------------------------------------------- /run_one2branch_qa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nju-websoft/One2Branch/HEAD/run_one2branch_qa.py -------------------------------------------------------------------------------- /run_one2seq_qa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nju-websoft/One2Branch/HEAD/run_one2seq_qa.py -------------------------------------------------------------------------------- /script_branch_kp.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nju-websoft/One2Branch/HEAD/script_branch_kp.sh -------------------------------------------------------------------------------- /script_branch_qa.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nju-websoft/One2Branch/HEAD/script_branch_qa.sh -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nju-websoft/One2Branch/HEAD/utils.py --------------------------------------------------------------------------------