├── .gitignore ├── LICENSE ├── NOTICE ├── README.md ├── SUMBT ├── LICENSE ├── README.md ├── code │ ├── BeliefTracker.py │ ├── BeliefTrackerSlotQueryMultiSlot.py │ ├── BeliefTrackerSlotQueryMultiSlotEWC.py │ ├── BeliefTrackerSlotQueryMultiSlotTransformer.py │ ├── BertForLabelEncoding.py │ ├── BertForUtteranceEncoding.py │ ├── __init__.py │ ├── main-multislot-ewc.py │ ├── main-multislot.py │ └── transform_augmented_data.py ├── requirements.txt ├── run-multiwoz.sh └── run.sh ├── __init__.py ├── assets ├── collector_labeler.png ├── fewshot_key.json ├── mapping.pair ├── ontology.json └── overview.png ├── create_data.py ├── neuralwoz ├── __init__.py ├── collector.py ├── demo.py ├── eval_labeler.py ├── goal_template.py ├── labeler.py ├── loss.py ├── models.py ├── preprocess.py ├── run_neuralwoz.py ├── synthesize │ ├── __init__.py │ ├── goal.py │ ├── mapping.pair │ ├── multiwoz_utils.py │ ├── override_utils.py │ └── value.py ├── train_collector.py ├── train_labeler.py └── utils │ ├── __init__.py │ ├── collector_utils.py │ ├── constants.py │ ├── data_utils.py │ ├── database.py │ ├── dream_utils.py │ ├── eval_utils.py │ ├── fix_label.py │ ├── labeler_utils.py │ └── seed.py ├── requirements.txt └── trade-dst ├── EWC_train.py ├── GEM_train.py ├── README.md ├── create_data.py ├── fine_tune.py ├── models └── TRADE.py ├── myTest.py ├── myTrain.py ├── nothing.txt ├── plot ├── HKUST.jpg ├── dataset.png ├── domain-slot.pdf ├── domain.pdf ├── einstein-scroll.png ├── example.png ├── model.png ├── pytorch-logo-dark.png ├── salesforce-research.jpg └── slot.pdf └── utils ├── config.py ├── fix_label.py ├── logger.py ├── mapping.pair ├── masked_cross_entropy.py ├── measures.py ├── multi-bleu.perl ├── utils_multiWOZ_DST.py └── utils_temp.py /.gitignore: -------------------------------------------------------------------------------- 1 | __pycache__ 2 | outputs/ 3 | data/ 4 | *env/ 5 | *.ipynb 6 | .DS_Store 7 | 8 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/LICENSE -------------------------------------------------------------------------------- /NOTICE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/NOTICE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/README.md -------------------------------------------------------------------------------- /SUMBT/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/SUMBT/LICENSE -------------------------------------------------------------------------------- /SUMBT/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/SUMBT/README.md -------------------------------------------------------------------------------- /SUMBT/code/BeliefTracker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/SUMBT/code/BeliefTracker.py -------------------------------------------------------------------------------- /SUMBT/code/BeliefTrackerSlotQueryMultiSlot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/SUMBT/code/BeliefTrackerSlotQueryMultiSlot.py -------------------------------------------------------------------------------- /SUMBT/code/BeliefTrackerSlotQueryMultiSlotEWC.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/SUMBT/code/BeliefTrackerSlotQueryMultiSlotEWC.py -------------------------------------------------------------------------------- /SUMBT/code/BeliefTrackerSlotQueryMultiSlotTransformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/SUMBT/code/BeliefTrackerSlotQueryMultiSlotTransformer.py -------------------------------------------------------------------------------- /SUMBT/code/BertForLabelEncoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/SUMBT/code/BertForLabelEncoding.py -------------------------------------------------------------------------------- /SUMBT/code/BertForUtteranceEncoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/SUMBT/code/BertForUtteranceEncoding.py -------------------------------------------------------------------------------- /SUMBT/code/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /SUMBT/code/main-multislot-ewc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/SUMBT/code/main-multislot-ewc.py -------------------------------------------------------------------------------- /SUMBT/code/main-multislot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/SUMBT/code/main-multislot.py -------------------------------------------------------------------------------- /SUMBT/code/transform_augmented_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/SUMBT/code/transform_augmented_data.py -------------------------------------------------------------------------------- /SUMBT/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/SUMBT/requirements.txt -------------------------------------------------------------------------------- /SUMBT/run-multiwoz.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/SUMBT/run-multiwoz.sh -------------------------------------------------------------------------------- /SUMBT/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/SUMBT/run.sh -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /assets/collector_labeler.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/assets/collector_labeler.png -------------------------------------------------------------------------------- /assets/fewshot_key.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/assets/fewshot_key.json -------------------------------------------------------------------------------- /assets/mapping.pair: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/assets/mapping.pair -------------------------------------------------------------------------------- /assets/ontology.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/assets/ontology.json -------------------------------------------------------------------------------- /assets/overview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/assets/overview.png -------------------------------------------------------------------------------- /create_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/create_data.py -------------------------------------------------------------------------------- /neuralwoz/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /neuralwoz/collector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/neuralwoz/collector.py -------------------------------------------------------------------------------- /neuralwoz/demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/neuralwoz/demo.py -------------------------------------------------------------------------------- /neuralwoz/eval_labeler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/neuralwoz/eval_labeler.py -------------------------------------------------------------------------------- /neuralwoz/goal_template.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/neuralwoz/goal_template.py -------------------------------------------------------------------------------- /neuralwoz/labeler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/neuralwoz/labeler.py -------------------------------------------------------------------------------- /neuralwoz/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/neuralwoz/loss.py -------------------------------------------------------------------------------- /neuralwoz/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/neuralwoz/models.py -------------------------------------------------------------------------------- /neuralwoz/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/neuralwoz/preprocess.py -------------------------------------------------------------------------------- /neuralwoz/run_neuralwoz.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/neuralwoz/run_neuralwoz.py -------------------------------------------------------------------------------- /neuralwoz/synthesize/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /neuralwoz/synthesize/goal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/neuralwoz/synthesize/goal.py -------------------------------------------------------------------------------- /neuralwoz/synthesize/mapping.pair: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/neuralwoz/synthesize/mapping.pair -------------------------------------------------------------------------------- /neuralwoz/synthesize/multiwoz_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/neuralwoz/synthesize/multiwoz_utils.py -------------------------------------------------------------------------------- /neuralwoz/synthesize/override_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/neuralwoz/synthesize/override_utils.py -------------------------------------------------------------------------------- /neuralwoz/synthesize/value.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/neuralwoz/synthesize/value.py -------------------------------------------------------------------------------- /neuralwoz/train_collector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/neuralwoz/train_collector.py -------------------------------------------------------------------------------- /neuralwoz/train_labeler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/neuralwoz/train_labeler.py -------------------------------------------------------------------------------- /neuralwoz/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /neuralwoz/utils/collector_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/neuralwoz/utils/collector_utils.py -------------------------------------------------------------------------------- /neuralwoz/utils/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/neuralwoz/utils/constants.py -------------------------------------------------------------------------------- /neuralwoz/utils/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/neuralwoz/utils/data_utils.py -------------------------------------------------------------------------------- /neuralwoz/utils/database.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/neuralwoz/utils/database.py -------------------------------------------------------------------------------- /neuralwoz/utils/dream_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/neuralwoz/utils/dream_utils.py -------------------------------------------------------------------------------- /neuralwoz/utils/eval_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/neuralwoz/utils/eval_utils.py -------------------------------------------------------------------------------- /neuralwoz/utils/fix_label.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/neuralwoz/utils/fix_label.py -------------------------------------------------------------------------------- /neuralwoz/utils/labeler_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/neuralwoz/utils/labeler_utils.py -------------------------------------------------------------------------------- /neuralwoz/utils/seed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/neuralwoz/utils/seed.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/requirements.txt -------------------------------------------------------------------------------- /trade-dst/EWC_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/trade-dst/EWC_train.py -------------------------------------------------------------------------------- /trade-dst/GEM_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/trade-dst/GEM_train.py -------------------------------------------------------------------------------- /trade-dst/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/trade-dst/README.md -------------------------------------------------------------------------------- /trade-dst/create_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/trade-dst/create_data.py -------------------------------------------------------------------------------- /trade-dst/fine_tune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/trade-dst/fine_tune.py -------------------------------------------------------------------------------- /trade-dst/models/TRADE.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/trade-dst/models/TRADE.py -------------------------------------------------------------------------------- /trade-dst/myTest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/trade-dst/myTest.py -------------------------------------------------------------------------------- /trade-dst/myTrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/trade-dst/myTrain.py -------------------------------------------------------------------------------- /trade-dst/nothing.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/trade-dst/nothing.txt -------------------------------------------------------------------------------- /trade-dst/plot/HKUST.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/trade-dst/plot/HKUST.jpg -------------------------------------------------------------------------------- /trade-dst/plot/dataset.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/trade-dst/plot/dataset.png -------------------------------------------------------------------------------- /trade-dst/plot/domain-slot.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/trade-dst/plot/domain-slot.pdf -------------------------------------------------------------------------------- /trade-dst/plot/domain.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/trade-dst/plot/domain.pdf -------------------------------------------------------------------------------- /trade-dst/plot/einstein-scroll.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/trade-dst/plot/einstein-scroll.png -------------------------------------------------------------------------------- /trade-dst/plot/example.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/trade-dst/plot/example.png -------------------------------------------------------------------------------- /trade-dst/plot/model.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/trade-dst/plot/model.png -------------------------------------------------------------------------------- /trade-dst/plot/pytorch-logo-dark.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/trade-dst/plot/pytorch-logo-dark.png -------------------------------------------------------------------------------- /trade-dst/plot/salesforce-research.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/trade-dst/plot/salesforce-research.jpg -------------------------------------------------------------------------------- /trade-dst/plot/slot.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/trade-dst/plot/slot.pdf -------------------------------------------------------------------------------- /trade-dst/utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/trade-dst/utils/config.py -------------------------------------------------------------------------------- /trade-dst/utils/fix_label.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/trade-dst/utils/fix_label.py -------------------------------------------------------------------------------- /trade-dst/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/trade-dst/utils/logger.py -------------------------------------------------------------------------------- /trade-dst/utils/mapping.pair: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/trade-dst/utils/mapping.pair -------------------------------------------------------------------------------- /trade-dst/utils/masked_cross_entropy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/trade-dst/utils/masked_cross_entropy.py -------------------------------------------------------------------------------- /trade-dst/utils/measures.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/trade-dst/utils/measures.py -------------------------------------------------------------------------------- /trade-dst/utils/multi-bleu.perl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/trade-dst/utils/multi-bleu.perl -------------------------------------------------------------------------------- /trade-dst/utils/utils_multiWOZ_DST.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/trade-dst/utils/utils_multiWOZ_DST.py -------------------------------------------------------------------------------- /trade-dst/utils/utils_temp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/naver-ai/neuralwoz/HEAD/trade-dst/utils/utils_temp.py --------------------------------------------------------------------------------