README.md 1.1 KB
Newer Older
W
wenquan wu 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46
Knowledge-driven Dialogue
=============================
[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)

This is a paddlepaddle implementation of generative-based model for knowledge-driven dialogue

## Requirements

* cuda=9.0
* cudnn=7.0
* python=2.7
* numpy
* paddlepaddle>=1.3.2

## Quickstart

### Step 1: Preprocess the data

Put the data of [DuConv](https://ai.baidu.com/broad/subordinate?dataset=duconv) under the data folder and rename them  train/dev/test.txt: 

```
./data/resource/train.txt
./data/resource/dev.txt
./data/resource/test.txt
```

### Step 2: Train the model

Train model with the following commands.

```bash
sh run_train.sh
```

### Step 3: Test the Model

Test model with the following commands.

```bash
sh run_test.sh
```

### Note !!!

* The script run_train.sh/run_test.sh shows all the processes including data processing and model training/testing. Be sure to read it carefully and follow it.
* The files in ./data and ./model is just empty file to show the structure of the document.