访问官网
Github
论文项目概述
Keras-TextClassification 是一个使用 Keras 库进行文本分类的开源项目。这个项目提供了一套丰富的文本分类算法，实现了从简单的模型到复杂的深度学习模型，包括 FastText、TextCNN、BERT、XLNet 等等。项目的目的是让开发者能够快速使用并对比不同的文本分类模型。
安装方式
要安装 Keras-TextClassification，可以使用 pip 命令：
pip install Keras-TextClassification
安装完成后，用户需要下载并解压必要的数据集，并将数据集复制到指定的路径中。
支持的模型
Keras-TextClassification 涵盖了多种文本分类模型：
- FastText: 一种高效的文本分类方法。
- TextCNN: 基于卷积神经网络的文本分类。
- TextRNN: 使用循环神经网络进行文本分类。
- BERT、XLNet、ALBERT 等预训练模型: 提供了处理自然语言的强大能力。
- 此外，还有其他诸如 TextRCNN、TextGCN、SelfAttention 以及 Transformer 编码器等模型。
使用说明
项目操作是通过运行 Python 脚本来完成的。以 FastText 为例，用户可以通过如下步骤执行：
- 进入
keras_textclassification/m01_FastText目录。 - 运行训练脚本：
python train.py - 运行预测脚本：
python predict.py
这里提供了不同模型的调用方式，以及如何进行多标签分类和句子相似度计算等。
数据集与准备
项目提供了部分示例数据集供用户使用：
- 百度问答语料集（baidu_qa_2019），以及今日头条新闻多标签数据集。
- 用户可以在项目中找到不同的预训练嵌入如 BERT、ALBERT、和 XLNet 的中文模型。
为了使用完整的数据集，用户可能需要通过提供的链接自行下载大量数据。
构建与设计
项目设计了简易的模块化架构：
- 基类负责构建网络结构和向量嵌入，其他模型继承基类。
- 提供了丰富的自定义层和预处理模块。
参考论文
以下是部分模型的相关论文标题：
- FastText: "Bag of Tricks for Efficient Text Classification"
- TextCNN: "Convolutional Neural Networks for Sentence Classification"
- BERT: "BERT: Pre-training of Deep Bidirectional Transformers for Language Understanding"
- XLNet: "XLNet: Generalized Autoregressive Pretraining for Language Understanding"
训练和调用示例
用户可以通过简单的脚本实现模型训练和测试。以下展示了一个使用 TextCNN 的训练调用示例：
from keras_textclassification import train
train(
graph='TextCNN',
label=17,
path_train_data=None,
path_dev_data=None,
rate=1,
hyper_parameters=None
)
Keras-TextClassification 项目通过提供多种模型和训练方式，加速了文本分类任务的开发和研究，致力于使文本分类在实际应用中更易于上手。
