Tested in Ubuntu 16.04
- Python 3.6.0
- Keras 2.0.5
- Tensorflow 1.3.0
- Theano 0.9.0
The /data
folder provides a small sample dataset for quick running.
To train and evaluate our model:
-
Download and unzip real dataset from Google Drive or Baidu Pan for Chinese users.
-
Replace each file in the
/data
folder with the corresponding real file.
Edit hyper-parameters and settings in config.py
python main.py --mode train
```bash
python main.py --mode eval