项目作者: liqiming-whu

项目描述 :
基于attention的CNN文本分类
高级语言: Python
项目地址: git://github.com/liqiming-whu/text-classification.git
创建时间: 2019-03-06T06:45:32Z
项目社区:https://github.com/liqiming-whu/text-classification

开源协议:

下载


基于attention的CNN文本分类

svm 基于tf-idf特征的支持向量机分类

att_cnn1 输入层attention

att_cnn2 卷积层attention

att_cnn3 双层attention

支持向量机

  1. python svm.py

训练

  1. python train.py --model base_cnn|att_cnn1|att_cnn2|att_cnn3 #default = att_cnn3

测试

  1. python test.py --model base_cnn|att_cnn1|att_cnn2|att_cnn3 #default = att_cnn3