文件名称：som-dst:SOM-DST
文件大小：264KB
文件格式：ZIP
更新时间：2024-04-10 10:40:49
dialogue dialogue-state-tracking Python
SOM-DST 此代码是进行pytorch正式实现。 ，， ，在ACL 2020中。 要求 python3.6 pytorch-transformers==1.0.0 torch==1.3.0a0+24ae9b5 wget==3.2 下载和预处理数据 要下载MultiWOZ数据集并对其进行预处理，请首先运行此脚本。您可以选择数据集的版本。 （“ 2.1”，“ 2.0”）下载的原始数据集将位于$DOWNLOAD_PATH ，经过预处理后，它将位于$TARGET_PATH 。 python3 create_data.py --main_dir $DOWNLOAD_PATH --target_path $TARGET_PATH --mwz_ver '2.1' # or '2.0' 模型训练 要训练SOM-DST模型，请运行此脚本。 $DATASET_DIR是预处理数据集的根目录，而$SAV
【文件预览】：
som-dst-master
----.gitignore(94B)
----README.md(5KB)
----utils()
--------eval_utils.py(2KB)
--------ckpt_utils.py(2KB)
--------data_utils.py(13KB)
--------__init__.py(0B)
--------fix_label.py(4KB)
--------mapping.pair(1KB)
----create_data.py(19KB)
----train.py(14KB)
----LICENSE(1KB)
----model.py(6KB)
----img()
--------overview6.png(142KB)
----requirements.txt(61B)
----evaluation.py(10KB)
----NOTICE(4KB)
----assets()
--------bert_config_large_uncased.json(314B)
--------bert_config_base_uncased.json(313B)
--------vocab.txt(226KB)