Skip to content
本项目为基于CNN,RNN 和NLP中预训练模型构建的多个常见的文本分类模型。
Jupyter Notebook Python
Branch: master
Clone or download

Latest commit

Fetching latest commit…
Cannot retrieve the latest commit at this time.

Files

Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
BERT
Bi-LSTM+Attention
Bi-LSTM
ELMo
RCNN
Transformer
adversarialLSTM
charCNN
dateHelper
images
textCNN
word2vec
.gitignore
README.md

README.md

文本分类项目


本项目为基于CNN,RNN 和NLP中预训练模型构建的多个常见的文本分类模型。

requirements

  • python==3.5.6
  • tensorflow-gpu==1.10.0

1. 数据集

1.1 IMDB电影评论的情感分析数据集

  数据集为cnews 文本分类数据集,总共有三个部分:

  • 带标签的训练集:../data/cnews_data/cnews.train.txt
  • 测试集:../data/cnews_data/cnews.train.txt

  字段的含义:

  • class 类别
  • review 内容
1.2 ChnSentiCorp 外卖评论情感分析数据集

  数据集为ChnSentiCorp

1.3 IMDB 电影评论数据集

  数据集为 IMDB

2. 数据预处理

  数据预处理方法/dataHelper/processData_*.ipynb

  将原始数据处理成干净的数据,处理后的数据存储在/data/preProcess下,数据预处理包括:

  • 去除各种标点符号
  • 生成训练word2vec模型的输入数据 /data/preProcess/wordEmbedding.txt

3. 训练word2vec词向量

  预训练word2vec词向量/word2vec/genWord2Vec*.ipynb

  • 预训练的词向量保存为bin格式 /word2vec/word2Vec*.bin

4. textCNN 文本分类

  textCNN模型来源于论文Convolutional Neural Networks for Sentence Classification

  textCNN可以看作是一个由三个单层的卷积网络的输出结果进行拼接的融合模型,作者提出了三种大小的卷积核[3, 4, 5],卷积核的滑动使得其 类似于NLP中的n-grams,因此当你需要更多尺度的n-grams时,你可以选择增加不同大小的卷积核,比如大小为2的卷积核可以代表 2-grams.

  textCNN代码在/textCNN/textCNN_*.ipynb。实现包括四个部分:

  • 参数配置类 Config (包括训练参数,模型参数和其他参数)
  • 数据预处理类 Dataset (包括生成词汇空间,获得预训练词向量,分割训练集和验证集)
  • textCNN模型类 TextCNN
  • 模型训练

5. charCNN 文本分类

  textCNN模型来源于论文Character-level Convolutional Networks for Text Classification

  char-CNN是一种基于字符级的文本分类器,将所有的文本都用字符表示, 注意这里的数据预处理时不可以去掉标点符号或者其他的各种符号,最好是保存论文中提出的69种字符,我一开始使用去掉特殊符号的字符后的文本输入到模型中会无法收敛。 此外由于训练数据集比较少,即使论文中最小的网络也无法收敛,此时可以减小模型的复杂度,包括去掉一些卷积层等。

  charCNN代码在/charCNN/charCNN.ipynb。实现也包括四个部分,也textCNN一致,但是在这里的数据预处理有很大不一样,剩下 的就是模型结构不同,此外模型中可以引入BN层来对每一层的输出做归一化处理。

6. Bi-LSTM 文本分类

  Bi-LSTM可以参考我的博客深度学习之从RNN到LSTM

  Bi-LSTM是双向LSTM,LSTM是RNN的一种,是一种时序模型,Bi-LSTM是双向LSTM,旨在同时捕获文本中上下文的信息, 在情感分析类的问题中有良好的表现。

  Bi-LSTM的代码在/Bi-LSTM/Bi-LSTM_*.ipynb中。除了模型类的代码有改动,其余代码几乎和textCNN一样。

7. Bi-LSTM + Attention 文本分类

  Bi-LSTM + Attention模型来源于论文Attention-Based Bidirectional Long Short-Term Memory Networks for Relation Classification

  Bi-LSTM + Attention 就是在Bi-LSTM的模型上加入Attention层,在Bi-LSTM中我们会用最后一个时序的输出向量 作为特征向量,然后进行softmax分类。Attention是先计算每个时序的权重,然后将所有时序 的向量进行加权和作为特征向量,然后进行softmax分类。在实验中,加上Attention确实对结果有所提升。

  Bi-LSTM + Attention的代码在/Bi-LSTM+Attention/Bi-LSTMAttention_*.ipynb中,除了模型类中 加入Attention层,其余代码和Bi-LSTM一致。

8. RCNN 文本分类

  RCNN模型来源于论文Recurrent Convolutional Neural Networks for Text Classification

  RCNN 整体的模型构建流程如下:

  • 利用Bi-LSTM获得上下文的信息,类似于语言模型
  • 将Bi-LSTM获得的隐层输出和词向量拼接[fwOutput, wordEmbedding, bwOutput]
  • 将拼接后的向量非线性映射到低维
  • 向量中的每一个位置的值都取所有时序上的最大值,得到最终的特征向量,该过程类似于max-pool
  • softmax分类

  RCNN的代码在/RCNN/RCNN_*.ipynb中。

9. adversarialLSTM 文本分类

  Adversarial LSTM模型来源于论文Adversarial Training Methods For Semi-Supervised Text Classification

  adversarialLSTM的核心思想是通过对word Embedding上添加噪音生成对抗样本,将对抗样本以和原始样本 同样的形式喂给模型,得到一个Adversarial Loss,通过和原始样本的loss相加得到新的损失,通过优化该新 的损失来训练模型,作者认为这种方法能对word embedding加上正则化,避免过拟合。

  adversarialLSTM的代码在/adversarialLSTM/adversarialLSTM_*.ipynb中。

10. Transformer 文本分类

  Transformer模型来源于论文Attention Is All You Need

  Transformer模型有两个结构:Encoder和Decoder,在进行文本分类时只需要用到 Encoder结构,Decoder结构是生成式模型,用于自然语言生成的。Transformer的核心结构是 self-Attention机制,具体的介绍见Transformer模型(Atention is all you need)

  Transformer模型的代码在/Transformer/transformer_*.ipynb中。

11. ELMo预训练模型 文本分类

  ELMo模型来源于论文Deep contextualized word representations

  ELMo的结构是BiLM(双向语言模型),基于ELMo的预训练模型能动态地生成 词的向量表示,具体的介绍见ELMO模型(Deep contextualized word representation)

  ELMo预训练模型用于文本分类的代码位于/ELMo/elmo_*.ipynb中。 /ELMo/bilm/下是ELMo项目中的源码,/ELMo/modelParams/下是各种文件。

12. Bert预训练模型 文本分类

  BERT模型来源于论文BERT: Pre-training of Deep Bidirectional Transformers for Language Understanding

  BERT模型是基于双向Transformer实现的语言模型,集预训练和下游任务于一个模型中, 因此在使用的时候我们不需要搭建自己的下游任务模型,直接用BERT模型即可,我们将谷歌开源的源码下载 下来放在bert文件夹中,在进行文本分类只需要修改run_classifier.py文件即可,另外我们需要将训练集 和验证集分割后保存在两个不同的文件中,放置在/BERT/data下。然后还需要下载谷歌预训练好的模型放置在 /BERT/modelParams文件夹下,还需要建一个/BERT/output文件夹用来放置训练后的模型文件

  做完上面的步骤之后只要执行下面的脚本即可

  export BERT_BASE_DIR=../modelParams/uncased_L-12_H-768_A-12

  export DATASET=../data/

  python run_classifier.py
    --data_dir=$MY_DATASET
    --task_name=imdb
    --vocab_file=$BERT_BASE_DIR/vocab.txt
    --bert_config_file=$BERT_BASE_DIR/bert_config.json
    --output_dir=../output/
    --do_train=true
    --do_eval=true
    --init_checkpoint=$BERT_BASE_DIR/bert_model.ckpt
    --max_seq_length=200
    --train_batch_size=16
    --learning_rate=5e-5
    --num_train_epochs=3.0

  BERT模型用于文本分类的详细使用可以看我的博客 文本分类实战(十)—— BERT 预训练模型

You can’t perform that action at this time.