├── .gitignore ├── LICENSE ├── README.md ├── db ├── attraction-dbase.db ├── attraction_db.json ├── attraction_db_processed.json ├── bus-dbase.db ├── bus_db.json ├── hospital-dbase.db ├── hospital_db.json ├── hospital_db_processed.json ├── hotel-dbase.db ├── hotel_db.json ├── hotel_db_processed.json ├── ontology.json ├── police_db.json ├── police_db_processed.json ├── restaurant-dbase.db ├── restaurant_db.json ├── restaurant_db_processed.json ├── taxi-dbase.db ├── taxi_db.json ├── taxi_db_processed.json ├── train-dbase.db ├── train_db.json ├── train_db_processed.json ├── value_set.json └── value_set_processed.json ├── galaxy ├── args.py ├── data │ ├── batch.py │ ├── data_loader.py │ ├── dataset.py │ ├── field.py │ ├── pretrain_field.py │ ├── sampler.py │ └── tokenizer.py ├── metrics │ ├── metrics.py │ └── metrics_tracker.py ├── models │ ├── __init__.py │ ├── generator.py │ ├── model_base.py │ ├── pretrain_unified_transformer.py │ └── unified_transformer.py ├── modules │ ├── embedder.py │ ├── feedforward.py │ ├── functions.py │ ├── multihead_attention.py │ └── transformer_block.py ├── trainers │ ├── pretrain_trainer.py │ └── trainer.py └── utils │ ├── clean_dataset.py │ ├── criterions.py │ ├── db_ops.py │ ├── eval.py │ ├── ontologies.py │ ├── ontology.py │ └── utils.py ├── preprocess.py ├── requirement.txt ├── run.py ├── run_pretrain.py ├── scripts ├── camrest │ ├── infer.sh │ └── train.sh ├── kvret │ ├── infer.sh │ └── train.sh ├── multiwoz2.0 │ ├── infer.sh │ └── train.sh ├── multiwoz2.1 │ ├── infer.sh │ └── train.sh └── pre_train │ ├── preprocess.sh │ ├── train_multi.sh │ └── train_single.sh └── tools └── mapping.pair /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/README.md -------------------------------------------------------------------------------- /db/attraction-dbase.db: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/db/attraction-dbase.db -------------------------------------------------------------------------------- /db/attraction_db.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/db/attraction_db.json -------------------------------------------------------------------------------- /db/attraction_db_processed.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/db/attraction_db_processed.json -------------------------------------------------------------------------------- /db/bus-dbase.db: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/db/bus-dbase.db -------------------------------------------------------------------------------- /db/bus_db.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/db/bus_db.json -------------------------------------------------------------------------------- /db/hospital-dbase.db: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /db/hospital_db.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/db/hospital_db.json -------------------------------------------------------------------------------- /db/hospital_db_processed.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/db/hospital_db_processed.json -------------------------------------------------------------------------------- /db/hotel-dbase.db: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/db/hotel-dbase.db -------------------------------------------------------------------------------- /db/hotel_db.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/db/hotel_db.json -------------------------------------------------------------------------------- /db/hotel_db_processed.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/db/hotel_db_processed.json -------------------------------------------------------------------------------- /db/ontology.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/db/ontology.json -------------------------------------------------------------------------------- /db/police_db.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/db/police_db.json -------------------------------------------------------------------------------- /db/police_db_processed.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/db/police_db_processed.json -------------------------------------------------------------------------------- /db/restaurant-dbase.db: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/db/restaurant-dbase.db -------------------------------------------------------------------------------- /db/restaurant_db.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/db/restaurant_db.json -------------------------------------------------------------------------------- /db/restaurant_db_processed.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/db/restaurant_db_processed.json -------------------------------------------------------------------------------- /db/taxi-dbase.db: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /db/taxi_db.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/db/taxi_db.json -------------------------------------------------------------------------------- /db/taxi_db_processed.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/db/taxi_db_processed.json -------------------------------------------------------------------------------- /db/train-dbase.db: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/db/train-dbase.db -------------------------------------------------------------------------------- /db/train_db.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/db/train_db.json -------------------------------------------------------------------------------- /db/train_db_processed.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/db/train_db_processed.json -------------------------------------------------------------------------------- /db/value_set.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/db/value_set.json -------------------------------------------------------------------------------- /db/value_set_processed.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/db/value_set_processed.json -------------------------------------------------------------------------------- /galaxy/args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/galaxy/args.py -------------------------------------------------------------------------------- /galaxy/data/batch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/galaxy/data/batch.py -------------------------------------------------------------------------------- /galaxy/data/data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/galaxy/data/data_loader.py -------------------------------------------------------------------------------- /galaxy/data/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/galaxy/data/dataset.py -------------------------------------------------------------------------------- /galaxy/data/field.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/galaxy/data/field.py -------------------------------------------------------------------------------- /galaxy/data/pretrain_field.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/galaxy/data/pretrain_field.py -------------------------------------------------------------------------------- /galaxy/data/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/galaxy/data/sampler.py -------------------------------------------------------------------------------- /galaxy/data/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/galaxy/data/tokenizer.py -------------------------------------------------------------------------------- /galaxy/metrics/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/galaxy/metrics/metrics.py -------------------------------------------------------------------------------- /galaxy/metrics/metrics_tracker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/galaxy/metrics/metrics_tracker.py -------------------------------------------------------------------------------- /galaxy/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/galaxy/models/__init__.py -------------------------------------------------------------------------------- /galaxy/models/generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/galaxy/models/generator.py -------------------------------------------------------------------------------- /galaxy/models/model_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/galaxy/models/model_base.py -------------------------------------------------------------------------------- /galaxy/models/pretrain_unified_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/galaxy/models/pretrain_unified_transformer.py -------------------------------------------------------------------------------- /galaxy/models/unified_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/galaxy/models/unified_transformer.py -------------------------------------------------------------------------------- /galaxy/modules/embedder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/galaxy/modules/embedder.py -------------------------------------------------------------------------------- /galaxy/modules/feedforward.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/galaxy/modules/feedforward.py -------------------------------------------------------------------------------- /galaxy/modules/functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/galaxy/modules/functions.py -------------------------------------------------------------------------------- /galaxy/modules/multihead_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/galaxy/modules/multihead_attention.py -------------------------------------------------------------------------------- /galaxy/modules/transformer_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/galaxy/modules/transformer_block.py -------------------------------------------------------------------------------- /galaxy/trainers/pretrain_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/galaxy/trainers/pretrain_trainer.py -------------------------------------------------------------------------------- /galaxy/trainers/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/galaxy/trainers/trainer.py -------------------------------------------------------------------------------- /galaxy/utils/clean_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/galaxy/utils/clean_dataset.py -------------------------------------------------------------------------------- /galaxy/utils/criterions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/galaxy/utils/criterions.py -------------------------------------------------------------------------------- /galaxy/utils/db_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/galaxy/utils/db_ops.py -------------------------------------------------------------------------------- /galaxy/utils/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/galaxy/utils/eval.py -------------------------------------------------------------------------------- /galaxy/utils/ontologies.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/galaxy/utils/ontologies.py -------------------------------------------------------------------------------- /galaxy/utils/ontology.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/galaxy/utils/ontology.py -------------------------------------------------------------------------------- /galaxy/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/galaxy/utils/utils.py -------------------------------------------------------------------------------- /preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/preprocess.py -------------------------------------------------------------------------------- /requirement.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/requirement.txt -------------------------------------------------------------------------------- /run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/run.py -------------------------------------------------------------------------------- /run_pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/run_pretrain.py -------------------------------------------------------------------------------- /scripts/camrest/infer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/scripts/camrest/infer.sh -------------------------------------------------------------------------------- /scripts/camrest/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/scripts/camrest/train.sh -------------------------------------------------------------------------------- /scripts/kvret/infer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/scripts/kvret/infer.sh -------------------------------------------------------------------------------- /scripts/kvret/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/scripts/kvret/train.sh -------------------------------------------------------------------------------- /scripts/multiwoz2.0/infer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/scripts/multiwoz2.0/infer.sh -------------------------------------------------------------------------------- /scripts/multiwoz2.0/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/scripts/multiwoz2.0/train.sh -------------------------------------------------------------------------------- /scripts/multiwoz2.1/infer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/scripts/multiwoz2.1/infer.sh -------------------------------------------------------------------------------- /scripts/multiwoz2.1/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/scripts/multiwoz2.1/train.sh -------------------------------------------------------------------------------- /scripts/pre_train/preprocess.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/scripts/pre_train/preprocess.sh -------------------------------------------------------------------------------- /scripts/pre_train/train_multi.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/scripts/pre_train/train_multi.sh -------------------------------------------------------------------------------- /scripts/pre_train/train_single.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/scripts/pre_train/train_single.sh -------------------------------------------------------------------------------- /tools/mapping.pair: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/siat-nlp/GALAXY/HEAD/tools/mapping.pair --------------------------------------------------------------------------------