├── .gitignore ├── LICENSE ├── README.md ├── compute_topic_signatures.py ├── demo.jsonl ├── planning ├── criterion.py ├── data │ ├── __init__.py │ ├── data_utils.py │ ├── dictionary.py │ ├── iterators.py │ └── text_planning_dataset.py ├── decode.py ├── models │ ├── __init__.py │ ├── bert_blocks.py │ └── bert_models.py ├── modules │ ├── __init__.py │ ├── adam.py │ ├── checkpoint_utils.py │ ├── inverse_square_root_scheduler.py │ ├── meters.py │ ├── progress_bar.py │ └── trainer.py ├── options.py ├── strategies │ ├── __init__.py │ ├── decoding_strategy.py │ └── strategy_utils.py ├── task.py ├── train.py ├── train_demo.sh ├── train_full.sh └── utils.py └── refinement ├── dataset ├── __init__.py ├── base_dataset.py ├── baseline_dataset.py └── refinement_dataset.py ├── decode_demo.sh ├── decoding_strategy ├── __init__.py ├── base_decoding.py ├── multipass_decoding.py ├── singlepass_decoding.py └── strategy_utils.py ├── generate.py ├── models ├── __init__.py ├── bart_blocks.py ├── bart_model.py ├── configuration.py └── modeling_utils.py ├── options.py ├── system.py ├── train.py ├── train_demo.sh └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XinyuHua/pair-emnlp2020/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XinyuHua/pair-emnlp2020/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XinyuHua/pair-emnlp2020/HEAD/README.md -------------------------------------------------------------------------------- /compute_topic_signatures.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XinyuHua/pair-emnlp2020/HEAD/compute_topic_signatures.py -------------------------------------------------------------------------------- /demo.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XinyuHua/pair-emnlp2020/HEAD/demo.jsonl -------------------------------------------------------------------------------- /planning/criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XinyuHua/pair-emnlp2020/HEAD/planning/criterion.py -------------------------------------------------------------------------------- /planning/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XinyuHua/pair-emnlp2020/HEAD/planning/data/__init__.py -------------------------------------------------------------------------------- /planning/data/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XinyuHua/pair-emnlp2020/HEAD/planning/data/data_utils.py -------------------------------------------------------------------------------- /planning/data/dictionary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XinyuHua/pair-emnlp2020/HEAD/planning/data/dictionary.py -------------------------------------------------------------------------------- /planning/data/iterators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XinyuHua/pair-emnlp2020/HEAD/planning/data/iterators.py -------------------------------------------------------------------------------- /planning/data/text_planning_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XinyuHua/pair-emnlp2020/HEAD/planning/data/text_planning_dataset.py -------------------------------------------------------------------------------- /planning/decode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XinyuHua/pair-emnlp2020/HEAD/planning/decode.py -------------------------------------------------------------------------------- /planning/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XinyuHua/pair-emnlp2020/HEAD/planning/models/__init__.py -------------------------------------------------------------------------------- /planning/models/bert_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XinyuHua/pair-emnlp2020/HEAD/planning/models/bert_blocks.py -------------------------------------------------------------------------------- /planning/models/bert_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XinyuHua/pair-emnlp2020/HEAD/planning/models/bert_models.py -------------------------------------------------------------------------------- /planning/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XinyuHua/pair-emnlp2020/HEAD/planning/modules/__init__.py -------------------------------------------------------------------------------- /planning/modules/adam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XinyuHua/pair-emnlp2020/HEAD/planning/modules/adam.py -------------------------------------------------------------------------------- /planning/modules/checkpoint_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XinyuHua/pair-emnlp2020/HEAD/planning/modules/checkpoint_utils.py -------------------------------------------------------------------------------- /planning/modules/inverse_square_root_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XinyuHua/pair-emnlp2020/HEAD/planning/modules/inverse_square_root_scheduler.py -------------------------------------------------------------------------------- /planning/modules/meters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XinyuHua/pair-emnlp2020/HEAD/planning/modules/meters.py -------------------------------------------------------------------------------- /planning/modules/progress_bar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XinyuHua/pair-emnlp2020/HEAD/planning/modules/progress_bar.py -------------------------------------------------------------------------------- /planning/modules/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XinyuHua/pair-emnlp2020/HEAD/planning/modules/trainer.py -------------------------------------------------------------------------------- /planning/options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XinyuHua/pair-emnlp2020/HEAD/planning/options.py -------------------------------------------------------------------------------- /planning/strategies/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XinyuHua/pair-emnlp2020/HEAD/planning/strategies/__init__.py -------------------------------------------------------------------------------- /planning/strategies/decoding_strategy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XinyuHua/pair-emnlp2020/HEAD/planning/strategies/decoding_strategy.py -------------------------------------------------------------------------------- /planning/strategies/strategy_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XinyuHua/pair-emnlp2020/HEAD/planning/strategies/strategy_utils.py -------------------------------------------------------------------------------- /planning/task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XinyuHua/pair-emnlp2020/HEAD/planning/task.py -------------------------------------------------------------------------------- /planning/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XinyuHua/pair-emnlp2020/HEAD/planning/train.py -------------------------------------------------------------------------------- /planning/train_demo.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XinyuHua/pair-emnlp2020/HEAD/planning/train_demo.sh -------------------------------------------------------------------------------- /planning/train_full.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XinyuHua/pair-emnlp2020/HEAD/planning/train_full.sh -------------------------------------------------------------------------------- /planning/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XinyuHua/pair-emnlp2020/HEAD/planning/utils.py -------------------------------------------------------------------------------- /refinement/dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XinyuHua/pair-emnlp2020/HEAD/refinement/dataset/__init__.py -------------------------------------------------------------------------------- /refinement/dataset/base_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XinyuHua/pair-emnlp2020/HEAD/refinement/dataset/base_dataset.py -------------------------------------------------------------------------------- /refinement/dataset/baseline_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XinyuHua/pair-emnlp2020/HEAD/refinement/dataset/baseline_dataset.py -------------------------------------------------------------------------------- /refinement/dataset/refinement_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XinyuHua/pair-emnlp2020/HEAD/refinement/dataset/refinement_dataset.py -------------------------------------------------------------------------------- /refinement/decode_demo.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XinyuHua/pair-emnlp2020/HEAD/refinement/decode_demo.sh -------------------------------------------------------------------------------- /refinement/decoding_strategy/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XinyuHua/pair-emnlp2020/HEAD/refinement/decoding_strategy/__init__.py -------------------------------------------------------------------------------- /refinement/decoding_strategy/base_decoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XinyuHua/pair-emnlp2020/HEAD/refinement/decoding_strategy/base_decoding.py -------------------------------------------------------------------------------- /refinement/decoding_strategy/multipass_decoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XinyuHua/pair-emnlp2020/HEAD/refinement/decoding_strategy/multipass_decoding.py -------------------------------------------------------------------------------- /refinement/decoding_strategy/singlepass_decoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XinyuHua/pair-emnlp2020/HEAD/refinement/decoding_strategy/singlepass_decoding.py -------------------------------------------------------------------------------- /refinement/decoding_strategy/strategy_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XinyuHua/pair-emnlp2020/HEAD/refinement/decoding_strategy/strategy_utils.py -------------------------------------------------------------------------------- /refinement/generate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XinyuHua/pair-emnlp2020/HEAD/refinement/generate.py -------------------------------------------------------------------------------- /refinement/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XinyuHua/pair-emnlp2020/HEAD/refinement/models/__init__.py -------------------------------------------------------------------------------- /refinement/models/bart_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XinyuHua/pair-emnlp2020/HEAD/refinement/models/bart_blocks.py -------------------------------------------------------------------------------- /refinement/models/bart_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XinyuHua/pair-emnlp2020/HEAD/refinement/models/bart_model.py -------------------------------------------------------------------------------- /refinement/models/configuration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XinyuHua/pair-emnlp2020/HEAD/refinement/models/configuration.py -------------------------------------------------------------------------------- /refinement/models/modeling_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XinyuHua/pair-emnlp2020/HEAD/refinement/models/modeling_utils.py -------------------------------------------------------------------------------- /refinement/options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XinyuHua/pair-emnlp2020/HEAD/refinement/options.py -------------------------------------------------------------------------------- /refinement/system.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XinyuHua/pair-emnlp2020/HEAD/refinement/system.py -------------------------------------------------------------------------------- /refinement/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XinyuHua/pair-emnlp2020/HEAD/refinement/train.py -------------------------------------------------------------------------------- /refinement/train_demo.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XinyuHua/pair-emnlp2020/HEAD/refinement/train_demo.sh -------------------------------------------------------------------------------- /refinement/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/XinyuHua/pair-emnlp2020/HEAD/refinement/utils.py --------------------------------------------------------------------------------