├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── LICENSE ├── README.md ├── __init__.py ├── compute_tree_acc.py ├── constrained_decoding ├── __init__.py ├── constrained_sequence_generator.py ├── constraint_checking.py └── fmttree.py ├── data ├── E2E_compositional │ ├── disc_test.tsv │ ├── test.tsv │ ├── testset_w_refs.csv │ ├── train.tsv │ └── val.tsv ├── weather │ ├── disc_test.tsv │ ├── test.tsv │ ├── train.tsv │ └── val.tsv └── weather_challenge │ ├── test.tsv │ ├── train.tsv │ └── val.tsv ├── example └── seq2seq_out.tsv ├── optim ├── __init__.py └── lr_scheduler │ ├── __init__.py │ └── reduce_lr_on_plateau.py ├── options.py ├── scripts ├── _eval_e2e_helper.py ├── count_failure_cases.sh ├── generate.e2e.lstm.sh ├── generate.weather.lstm.sh ├── generate.weather_challenge.lstm.sh ├── measure_scores.e2e.replfail.sh ├── measure_scores.e2e.sh ├── measure_scores.replfail.sh ├── measure_scores.sh ├── prepare.e2e.sh ├── prepare.weather.sh ├── prepare.weather_challenge.sh ├── train.e2e.lstm.sh ├── train.weather.lstm.sh ├── train.weather_challenge.lstm.sh ├── tree_acc.replfail.sh └── tree_acc.sh └── tasks ├── __init__.py └── fairseq_task.py /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/TreeNLG/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/TreeNLG/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/TreeNLG/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/TreeNLG/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/TreeNLG/HEAD/__init__.py -------------------------------------------------------------------------------- /compute_tree_acc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/TreeNLG/HEAD/compute_tree_acc.py -------------------------------------------------------------------------------- /constrained_decoding/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/TreeNLG/HEAD/constrained_decoding/__init__.py -------------------------------------------------------------------------------- /constrained_decoding/constrained_sequence_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/TreeNLG/HEAD/constrained_decoding/constrained_sequence_generator.py -------------------------------------------------------------------------------- /constrained_decoding/constraint_checking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/TreeNLG/HEAD/constrained_decoding/constraint_checking.py -------------------------------------------------------------------------------- /constrained_decoding/fmttree.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/TreeNLG/HEAD/constrained_decoding/fmttree.py -------------------------------------------------------------------------------- /data/E2E_compositional/disc_test.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/TreeNLG/HEAD/data/E2E_compositional/disc_test.tsv -------------------------------------------------------------------------------- /data/E2E_compositional/test.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/TreeNLG/HEAD/data/E2E_compositional/test.tsv -------------------------------------------------------------------------------- /data/E2E_compositional/testset_w_refs.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/TreeNLG/HEAD/data/E2E_compositional/testset_w_refs.csv -------------------------------------------------------------------------------- /data/E2E_compositional/train.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/TreeNLG/HEAD/data/E2E_compositional/train.tsv -------------------------------------------------------------------------------- /data/E2E_compositional/val.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/TreeNLG/HEAD/data/E2E_compositional/val.tsv -------------------------------------------------------------------------------- /data/weather/disc_test.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/TreeNLG/HEAD/data/weather/disc_test.tsv -------------------------------------------------------------------------------- /data/weather/test.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/TreeNLG/HEAD/data/weather/test.tsv -------------------------------------------------------------------------------- /data/weather/train.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/TreeNLG/HEAD/data/weather/train.tsv -------------------------------------------------------------------------------- /data/weather/val.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/TreeNLG/HEAD/data/weather/val.tsv -------------------------------------------------------------------------------- /data/weather_challenge/test.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/TreeNLG/HEAD/data/weather_challenge/test.tsv -------------------------------------------------------------------------------- /data/weather_challenge/train.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/TreeNLG/HEAD/data/weather_challenge/train.tsv -------------------------------------------------------------------------------- /data/weather_challenge/val.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/TreeNLG/HEAD/data/weather_challenge/val.tsv -------------------------------------------------------------------------------- /example/seq2seq_out.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/TreeNLG/HEAD/example/seq2seq_out.tsv -------------------------------------------------------------------------------- /optim/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /optim/lr_scheduler/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /optim/lr_scheduler/reduce_lr_on_plateau.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/TreeNLG/HEAD/optim/lr_scheduler/reduce_lr_on_plateau.py -------------------------------------------------------------------------------- /options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/TreeNLG/HEAD/options.py -------------------------------------------------------------------------------- /scripts/_eval_e2e_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/TreeNLG/HEAD/scripts/_eval_e2e_helper.py -------------------------------------------------------------------------------- /scripts/count_failure_cases.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/TreeNLG/HEAD/scripts/count_failure_cases.sh -------------------------------------------------------------------------------- /scripts/generate.e2e.lstm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/TreeNLG/HEAD/scripts/generate.e2e.lstm.sh -------------------------------------------------------------------------------- /scripts/generate.weather.lstm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/TreeNLG/HEAD/scripts/generate.weather.lstm.sh -------------------------------------------------------------------------------- /scripts/generate.weather_challenge.lstm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/TreeNLG/HEAD/scripts/generate.weather_challenge.lstm.sh -------------------------------------------------------------------------------- /scripts/measure_scores.e2e.replfail.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/TreeNLG/HEAD/scripts/measure_scores.e2e.replfail.sh -------------------------------------------------------------------------------- /scripts/measure_scores.e2e.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/TreeNLG/HEAD/scripts/measure_scores.e2e.sh -------------------------------------------------------------------------------- /scripts/measure_scores.replfail.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/TreeNLG/HEAD/scripts/measure_scores.replfail.sh -------------------------------------------------------------------------------- /scripts/measure_scores.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/TreeNLG/HEAD/scripts/measure_scores.sh -------------------------------------------------------------------------------- /scripts/prepare.e2e.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/TreeNLG/HEAD/scripts/prepare.e2e.sh -------------------------------------------------------------------------------- /scripts/prepare.weather.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/TreeNLG/HEAD/scripts/prepare.weather.sh -------------------------------------------------------------------------------- /scripts/prepare.weather_challenge.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/TreeNLG/HEAD/scripts/prepare.weather_challenge.sh -------------------------------------------------------------------------------- /scripts/train.e2e.lstm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/TreeNLG/HEAD/scripts/train.e2e.lstm.sh -------------------------------------------------------------------------------- /scripts/train.weather.lstm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/TreeNLG/HEAD/scripts/train.weather.lstm.sh -------------------------------------------------------------------------------- /scripts/train.weather_challenge.lstm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/TreeNLG/HEAD/scripts/train.weather_challenge.lstm.sh -------------------------------------------------------------------------------- /scripts/tree_acc.replfail.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/TreeNLG/HEAD/scripts/tree_acc.replfail.sh -------------------------------------------------------------------------------- /scripts/tree_acc.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/TreeNLG/HEAD/scripts/tree_acc.sh -------------------------------------------------------------------------------- /tasks/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tasks/fairseq_task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/TreeNLG/HEAD/tasks/fairseq_task.py --------------------------------------------------------------------------------