Skip to content

Embedding, NMT, Text_Classification, Text_Generation, NER etc.

Notifications You must be signed in to change notification settings

shawroad/NLP_pytorch_project

Repository files navigation

注意

由于目前这个仓库项目过多,不易维护。因此,后续将各种任务独立开来,分别建立仓库,做很多对比实验,方便使用者挑选模型。有需要的,可以跳转指定仓库,链接如下:

目录

NLP_pytorch_project

Chatbot

1. Bert_chatbot: 类似于UniLM的方式

  • python train.py   # 训练代码
  • python infernece.py   # 模型的推理

2. seq2seq_luong: 编码器为两层gru网络,解码器为一层gru网络,在编码器和解码器中间,加入了luong注意力。

  • python train.py   # 训练代码
  • python inference.py   # 模型的推理

3. transformer_chatbot: 标准的transformer模型

  • python train.py   # 训练代码
  • python chat.py   # 可以进行聊天 训练数据采用的是青云对话语料

Distillation

1. DynaBert: 华为的工作,主要是采用剪枝的方式对bert的某些结构进行裁剪。

  • python train_teacher_model.py   # 训练老师模型
  • python train_tailor_model.py   # 对老师模型剪枝

2. rnn_distill_bert: 用一层lstm网络去蒸馏bert模型,只加入了软标签损失。

  • python train_bert.py   # 训练老师模型 bert
  • python train_distill.py   # 蒸馏 用lstm学习bert的输出

3. three_layer_self-attention_to_distill_bert: 看名字大概都知道啥意思了,就是写了三层transformer的encoder,然后去蒸馏bert模型

  • python train_bert.py   # 训练老师模型bert
  • python train_distill.py   # 蒸馏

4. tiny_bert: 华为的工作,tiny_bert的蒸馏方式是除了加入软标签损失,还加入了中间层的均方误差损失

  • python train.py   # 训练老师模型 bert
  • python train_distill_v2.py   # 蒸馏

Embedding

1. skipgram-word2vec: 使用skipgram的方式得到词向量

  • python 001-skipgram-word2vec.py

2. bert: 直接训练bert, 从头训练, 也可以使用此代码进行再训练

  • python 002-bert.py

3. albert: 直接训练albert,从头训练, 也可以使用此代码进行再训练

  • python 003-albert.py

4. NPLM: 传统方法

  • python 004-NPLM.py

NER

1. Bert_CRF_Ner: bert模型加条件随机场进行序列标注任务

  • python run_ner_crf.py   # 模型的训练
  • python inference.py   # 模型的推理

2. Bert_Softmax_Ner: 直接使用bert模型进行序列标注

  • python train.py   # 模型的训练
  • python inference.py   # 模型的推理

3. BiLSTM_CRF_Ner: 使用双向的lstm网络和crf进行序列标注任务

  • python train.py   # 模型的训练

NMT

1. GRU_attention: 编码器和解码器都是gru网络,中间加入一个普通的注意力机制(直接的加权和)

  • python train.py   # 模型的训练

2. Transformers_NMT: 标准的transformer结构做机器翻译

  • python train.py   # 模型的训练

Pretrain_Model

1. bert-pretrain: bert模型的再训练,首先通过get_train_data.py进行数据预处理,包含15%的词进行mask等操作,然后在进行训练。

  • python get_train_data.py   # 数据预处理
  • python run_pretrain.py   # 再训练

2. wobert-pretrain: wobert的预训练模型由苏神给出,这里的再训练可以加入自己构建的词表,然后修改了bert的分词方式。

  • python process_pretrain_data.py   # 数据预处理
  • python run_pretrain.py   # 再训练

Reading_comprehension

1. BERT_MRC: 使用bert去做机器阅读理解任务。这里长预料采用直接阶段的方式。

  • python train.py   # 模型训练

2. BiDAF: 双向注意力流机制的机器阅读理解模型

  • python data_process.py   # 首先对数据预处理
  • python train_bidaf.py   # 模型的训练

3. DocQA: 传统的模型

  • python data_process.py   # 首先对数据预处理
  • python train_DocQA.py   # 模型的训练

4. Match_LSTM: 传统的模型,单纯的rnn结构。

  • python data_process.py   # 首先对数据预处理
  • python train_Match_Lstm.py   # 模型的训练

5. QANet: 也是一个比较传统的模型,但是这个模型是第一个抛弃rnn结构的mrc模型,这个模型也是第一个将self-attention机制引入到mrc任务中的。

  • python data_process.py   # 数据预处理
  • python train.py   # 模型的训练

6. RNet: 传统的模型

  • python data_process.py   # 数据预处理
  • python train_RNet.py   # 模型的训练

7. Recurrence-hotpot-baseline: 首次产生用rnn结构处理多跳推理的问题,在hotpotqa数据集中,除了包含答案的预测,还有支撑事实的预测,以及相关段落的预测。

  • python data_process.py   # 数据预处理
  • python train.py   # 模型的训练

8. albert_mrc: 使用albert预训练模型去做mrc任务

  • python train_update.py   # 训练模型
  • python inference.py   # 单条数据的推理
  • python inference_all.py   # 所有数据的推理

9. electra_bert: 使用electra预训练模型去做mrc任务

  • python run_cail.py   # 训练模型
  • python evaluate.py   # 模型的评估

10. mrc_baseline: 如果做mrc任务,建议首先看这个代码,这里面包含了mrc注意的各种细节,如长文本的处理(滑动窗口), 答案的排序,对抗训练等等。

  • python train.py   # 训练模型

11. roberta_mrc: 使用roberta预训练模型去做mrc任务

  • python train.py   # 训练模型

12. transformer+rnn+attention: 这个项目是做生成式的阅读理解,直接采用的是seq2seq的方式,编码器采用的是transformer的encoder, 解码器采用的gru的结构,中间还加入了一层普通的注意力机制。

  • python train.py   # 模型的训练
  • python inference.py   # 模型的推理

13. transformer_reading: 这个项目也是做生成式阅读理解,采用的是标准的transformer结构。

  • python train.py   # 模型的训练
  • python inference.py   # 模型的推理

Slot_Filling

1. JointBert: 涉及到意图分类和槽分类。直接用bert对输入进行编码,用“CLS”向量进行意图分类。用每个token的最终编码向量进行槽分类。

  • python train.py   # 对模型的训练

Text_Classification

1. DPCNN: 深层的卷积网络+残差的连接使得此模型比以往的CNN结构都好,并且复杂度也不高。

  • python get_data_to_examples.py   # 预处理数据
  • python examples_to_features.py   # 将对应的example转为feature
  • python train.py   # 模型的训练

2. FastBert: 采用一种自蒸馏的方式加快模型的推理。主要应用在分类任务上。

  • sh train_stage0.sh   # 训练老师模型 bert
  • sh train_stage1.sh   # 自蒸馏
  • sh infer_sigle.sh   # 自适应的推理单条样本

3. FastText: 由facebook提出,是一种高效的文本分类模型。

  • python step1_get_data_to_examples.py   # 获取数据
  • python step2_examples_to_features.py   # 将文本数据转为id序列
  • python train.py   # 模型训练

4. XLNet: 1)通过最大化所有可能的因式分解顺序的对数似然,学习双向语境信息;2)用自回归本身的特点克服 BERT 的缺点。此外,XLNet 还融合了当前最优自回归模型 Transformer-XL 的思路。

  • python train.py   # 模型训练

5. all_layer_out_concat: 从名字上就可以看出,本项目是将文本经过bert-style模型进行编码,然后将每层的cls向量拿出来,进行一个注意力计算,然后进行分类。

  • python train.py   # 模型训练
  • Python inference.py   # 模型推理

6. bert+bceloss+average_checkpoint: 该项目将分类任务的损失函数改成了BCELoss,另外,加入的权重平均(将多个checkpoint进行平均)

  • python run_classify.py   # 模型训练
  • python run_average_checkpoints.py   # 权重平均

7. capsule_text_classification: GRU+Capsule进行文本分类

  • python train.py   # 模型训练

8. longformer_classification: 使用预训练模型longformer进行文本分类。对于长文本的分类,可以试试这个模型。

  • python train.py   # 模型训练

9. multi_label_classify_bert: 使用bert模型进行多标签分类。里面包含三个模型: bert(model.py)、bert后两层池化(model2.py)、bert+TextCNN(model3.py)。

  • python train.py   # 模型训练
  • python inference.py   # 模型预测

10. roberta_classification: 使用roberta预训练模型进行文本分类。

  • python train.py   # 模型训练

11. transformer_xl: 直接使用transformer_xl进行文本分类,对于长文本分类,可以试试此模型。

  • python train.py   # 模型训练

12. wobert+focal_loss: wobert预训练模型有苏神提供,在分类任务中加入了focal loss解决类别不平衡的问题。

  • python run_classify.py   # 模型训练

13. TextCNN: 对文本进行不同尺度的卷积,然后拼接进行文本分类。

  • python 001-TextCNN.py   # 模型训练

14. BILSTM+Attention: 双向的LSTM网络加普通的注意力进行文本分类。

  • python 002-BILSTM+Attention.py   # 模型训练

Text_Clustering

1. LDA聚类

  • python train_LDA_cluster.py   # 聚类

2. DBSCAN

  • python train_dbscan_cluster.py   # 聚类

3. Kmeans

  • python train_kmeans_cluster.py   # 聚类

Text_Corrector

1. bert_for_correction: 只是简单的尝试,在对应的语料上进行了再训练,输入一个有错别字的句子,然后对每个token的编码向量进行分类。

  • python run_pretrain_bert.py   # 再训练
  • bert_corrector.py   # 纠错

Text_Generation

1. GPT2_SummaryGen: 使用GPT2进行摘要的生成

  • python train.py   # 模型训练
  • python inferface.py   # 模型推理

2. GPT2_TitleGen: 文章标题的生成

  • python train.py   # 模型训练
  • python inference.py   # 模型推理

3. Simple-GPT2: 自己实现的GPT2模型

  • python train.py   # 模型训练
  • python inference.py   # 模型推理

Text_Ranking

1. BM25: 计算query与所有待排序文本的BM25值,然后根据这个值进行排序。

  • python main.py   # 排序

2. DC_Bert_Ranking: 双塔+交互。首先对query和context分别进行编码,此处的权重不共享,然后将query和context的编码混合,再通过几层交互的transformer-encoder。

  • python train.py   # 模型训练
  • python inference.py   # 模型推理

3. DPR_Ranking:Facebook的文本排序模型

  • python train.py   # 模型训练

4. MT_Ranking: 用bert-style模型进行编码,然后用cls进行分类,按正样本的得分排序

  • python train.py   # 模型训练
  • python inference.py   # 模型推理

5. ReRank: 含模型蒸馏

  • python train.py   # 模型训练
  • python train_distill.py   # 模型蒸馏

Text_Similarity

1. ABCNN: 首先对两个句子进行词嵌入,然后进行池化,得到两个句子的向量,接着计算两个向量的差值,最后对这个差值向量进行不同尺度的卷积,然后进行分类。

  • python train.py   # 模型训练

2. BiMPM: 该模型分为四部分: 词嵌入、上下文编码(bilstm)、四种Matching、Aggregation Layer。

  • python train.py   # 模型训练

3. DecomposableAttention:这篇论文的核心就是alignment,即词与词的对应关系,文中的alignment用在了两个地方,一个attend部分,是用来计算两个句子之间的attention关系,另一个是compare部分,对两个句子之间的词进行比较,每次的处理都是以词为单位的,最后用前馈神经网络去做预测。很明显可以看到,这篇文章提到的模型并没有使用到词在句子中的时序关系,更多的是强调两句话的词之间的对应关系(alignment)

  • python train.py   # 模型训练

4. ESIM: 短文本匹配利器。ESIM 牛逼在它的 inter-sentence attention,就是代码中的 soft_align_attention,这一步中让要比较的两句话产生了交互。以前类似 Siamese 网络的结构,往往中间都没有交互,只是在最后一层求个余弦距离或者其他距离。

  • python train.py   # 模型训练

5. RE2: RE2 这个名称来源于该网络三个重要部分的合体:Residual vectors;Embedding vectors;Encoded vectors。

  • python train.py   # 模型训练

6. SiaGRU: 双塔结构,对两个sentence分别用GRU编码,然后计算两个句子编码向量的差值,最后用这个差值向量进行分类。

  • python train.py   # 模型训练

7. SimCSE: 对比学习,技巧:不同的样本不同的Dropout

  • python train.py   # 模型训练

8. BM25: 直接计算两个文本的BM25值,代表其相似程度。

  • python BM25.py

9. TF_IDF: 直接计算两个文本的TF_IDF值,代表其相似程度。

  • python TF_IDF.py

10. NEZHA_Coattention: 采用的是双塔结构,将两个句子分别输入到NEZHA模型中,然后将输入做差,再与原来的表示拼接,接着送到一个全连接网络中进行分类。还有一个模型,就是得到两个句子的表示后,我们自己实现了一层transformer-encoder,进行表示信息的融合,最后再进行分类。

  • python train.py

11. Bert_Whitening: 方法由苏剑林提出,无需训练,直接将每个句子的bert输出统一到标准正交基的下

  • python run_bert_whitening.py # 直接对数据集进行验证 计算斯皮尔曼系数

data_augmentation

1. eda: 使用nlpcda工具包进行数据增广。如: 等价实体替换、随机同义词替换、随机删除字符、位置交换、同音词替换。

  • python 001-run_eda.py

2. 回译-百度: 使用百度翻译进行文本的回译。

  • python 002-run_contrslate_data_aug.py

3. 回译-google: 使用google翻译进行文本的回译。

  • python 003-google_trans_data_aug.py

relation_extraction

1. lstm_cnn_information_extract: lstm+cnn

  • python train.py   # 模型训练
  • python inference.py   # 模型推理

2. relation_classification: 关系分类, bilstm+普通注意力

  • python data_helper.py   # 数据预处理
  • python train.py   # 模型训练

Star History

Star History Chart

Releases

No releases published

Packages

No packages published

Languages