EDGE exemplars:论文代码 使用语义示例控制对话生成(Naacl 2021)基于语义示例的检索 优化方法用于对话响应生成 源码

EDGE范例 论文代码-使用语义样本控制对话生成(NAACL 2021)一种基于语义样本的用于对话响应生成的检索-精简方法 数据处理 用于数据用自述沿处理码和示例文件存在于文件夹data_processing 模型训练 要训​​练EDGE模型,请运行train_robust.py文件python train_robust.py --dataset_path $LOCATION_OF_JSON_DATA 要生成测试json的响应,您可以同时运行脚本run_generate.py或其多处理版本run_generate_batch.py python run_generate.py --model_checkpoint $TRAINED_MODEL_FOLDER --dataset_path $LOCATION_OF_JSON_DATA python run_generate_batch.py