siamese_cnn

Siamese卷积神经网络完成问答任务

基于Tensorflow和WikiQA数据集。

利用卷积层和池化层实现的一个pointwise的QA网络。

依赖

• python2.7
• tensorflow==1.8.0

安装

如果使用本书附带的docker镜像，所有依赖已经安装好，不需要再次安装。使用docker镜像运行程序的方式详见文档。

下载词向量文件glove。

../download.sh

预处理wiki数据

../preprocess.sh

正常运行后，有下面输出：

运行

训练模型

./train.sh

测试模型

./test.sh

某模型测试结果

root@feedd41cb9b4:/app/ch4/siamese_cnn# ./test.sh
test model
embedding file: /tools/embedding/glove.6B.100d.txt
Pre-trained: 27058 (92.94%)
[test] MAP:0.642138055098, MRR:0.642658016961