Skip to content

Latest commit

 

History

History
40 lines (32 loc) · 1.56 KB

Readme.md

File metadata and controls

40 lines (32 loc) · 1.56 KB
Raw

DyKGChat

The project contains the collected data and code of our paper Yi-Lin Tuan, Yun-Nung Chen, Hung-yi Lee. "DyKgChat: Benchmarking Dialogue Generation Grounding on Dynamic Knowledge Graphs", EMNLP 2019.

The extended abstract version is called Dynamic Knowledge-Grounded Dialogue Generation through Walking on the Graph.

  • our proposed approach: (Qadpt) Quick Adaptive Dynamic Knoledge-Grounded Neural Converation Model (pronouce: Q-adapt)

Qadpt

Setup

Installation (my environment)

  • python3.6
  • tensorflow r1.13
  • jieba
  • nltk3.2.5

Files

  • data/: the collected data hgzhz/ and friends/ as well as the trained TransE
  • model_ckpts/: the trained models in the paper

Usage

  • clone the repository
  • run the script run.sh
$bash run.sh <GPU_ID> <method> <model> <data> <exp_name>
  • for <GPU_ID>, check your device avalibility by nvidia-smi
  • for , choose from train, pred_acc, eval_pred_acc, ifchange
  • for , choose from seq2seq, MemNet, TAware, KAware, Qadpt
  • for , choose from friends, hgzhz_v1_0(used in our paper), hgzhz(current newest version)
  • for <exp_name>, check the directory model_ckpts

More description

  • testing method
    • pred_acc: for metrics Generated-KW, BLEU-2, distinct-n
    • eval_pred_acc: for metrics KW-Acc, KW/Generic, perplexity
    • ifchange: for change rates / accurate change rates
  • script options
    • the hops_num and change_level are required to be changed in run.sh