- gensim
pip install gensim
- Tensorflow 1.9-1.12 + Python2.7
-
Download the Ubuntu dataset released by (Xu et al, 2017)
-
Unzip the dataset and put data directory into
data/
-
Preprocess dataset, including concatenatate context and build vocabulary
cd data
python prepare.py
- Train word2vec
bash run_train_word2vec.sh
- Train and test ESIM, the log information is in
log.txt
file. You could find an example log file inlog_example.txt
.
cd scripts/esim
bash run.sh
from https://github.com/alibaba/esim-response-selection
No comments:
Post a Comment