本次只实现3*3的卷积核
修改构建图的code:
1.因为换了模型,所以需要换参数:
删掉:
添加参数:
2.更改计算图的构建:
注释掉lstm的实现:
实现TextCNN:
定义初始化函数:
开一个新的运行空间:
将max_pooling层接到fc上:
修改train_op:
直接用optiliarzer即可,不用做clip:
3.运行效果:
可以发现TextCNN训练速度比LSTM快很多,但是TextCNN在输入上不能是任意长度的