TripPy:用于值独立神经对话框状态跟踪的三重复制策略
该代码是其的改编。 在这里,我们仅包括对整个MultiWOZ数据集的训练以及对CoCo生成的数据的再训练。
数据
MultiWOZ数据,将其解压缩,然后将结果data文件夹放置到当前目录( coco-dst/trippy-public/ )中。
训练基线
运行以下命令以启动基线模型训练:
❱❱❱ sh train_baseline.sh
训练完模型后,将interface4eval_trippy.py第50行MODEL_CHECKPOINT修改为要评估的检查点的路径。 您也可以我们在本文中使用的预先训练的检查点。
然后,您可以通过运行以下命令,在CoCo生成的示例上简单地评估所需的TripPy模型检查点
❱❱❱ sh run_eval.sh
按照../coco-dst/README.md Run Evaluation on
2021-08-23 11:24:28
51KB
Python
1