Migrated repository
Go to file
2019-08-16 17:40:16 +08:00
.github Update PULL_REQUEST_TEMPLATE.md 2018-12-14 19:10:44 +08:00
docs update docs-tools 2019-08-16 17:40:16 +08:00
fastNLP update docs of io.file_utils 2019-08-16 16:53:02 +08:00
legacy fix bug in load dataset test code 2019-07-09 02:18:11 +08:00
reproduction merge docs 2019-08-14 23:16:11 +08:00
test 1. Trainer增加一个dev_batch_size参数;2.StaticEmbedding中增加min_freq; 2019-08-16 02:19:23 +08:00
tutorials Dev0.4.0 (#149) 2019-05-22 18:43:56 +08:00
.gitignore Create .gitignore 2019-06-05 16:35:59 +08:00
.travis.yml travis.yml格式错误,修正 2019-08-16 00:58:56 +08:00
codecov.yml add codecov fix 2019-02-04 09:56:08 +08:00
LICENSE add LICENSE, setup.py & requirements.txt 2018-05-25 18:32:02 +08:00
MANIFEST.in Dev0.4.0 (#149) 2019-05-22 18:43:56 +08:00
README.md 在README.md添加版本说明 2019-07-12 14:12:13 +08:00
readthedocs.yml 修改了最新的文档 2019-05-24 01:58:04 +08:00
requirements.txt 删除elmo对h5py的依赖 2019-07-09 17:32:22 +08:00
setup.py 1.修复trainer中潜在多步更新bug; 2. LSTM的数据并行修改;3. embed_loader中bug修复, 且允许手动初始化; 2019-06-30 09:52:01 +08:00

fastNLP

Build Status codecov Pypi Hex.pm Documentation Status

fastNLP 是一款轻量级的 NLP 处理套件。你既可以使用它快速地完成一个序列标注(NER、POS-Tagging等、中文分词、文本分类Matching指代消解摘要等任务; 也可以使用它构建许多复杂的网络模型,进行科研。它具有如下的特性:

  • 统一的Tabular式数据容器让数据预处理过程简洁明了。内置多种数据集的DataSet Loader省去预处理代码;
  • 多种训练、测试组件例如训练器Trainer测试器Tester以及各种评测metrics等等;
  • 各种方便的NLP工具例如预处理embedding加载包括ELMo和BERT; 中间数据cache等;
  • 详尽的中文文档教程以供查阅;
  • 提供诸多高级模块例如Variational LSTM, Transformer, CRF等;
  • 在序列标注、中文分词、文本分类、Matching、指代消解、摘要等任务上封装了各种模型可供直接使用详细内容见 reproduction 部分;
  • 便捷且具有扩展性的训练器; 提供多种内置callback函数方便实验记录、异常捕获等。

安装指南

fastNLP 依赖以下包:

  • numpy>=1.14.2
  • torch>=1.0.0
  • tqdm>=4.28.1
  • nltk>=3.4.1
  • requests
  • spacy

其中torch的安装可能与操作系统及 CUDA 的版本相关,请参见 PyTorch 官网 。 在依赖包安装完成后,您可以在命令行执行如下指令完成安装

pip install fastNLP
python -m spacy download en

目前使用pip安装fastNLP的版本是0.4.1有较多功能仍未更新最新内容以master分支为准。 fastNLP0.5.0版本将在近期推出,请密切关注。

fastNLP教程

内置组件

大部分用于的 NLP 任务神经网络都可以看做由词嵌入embeddings和两种模块编码器encoder、解码器decoder组成。

以文本分类任务为例下图展示了一个BiLSTM+Attention实现文本分类器的模型流程图

fastNLP 在 embeddings 模块中内置了几种不同的embedding静态embeddingGloVe、word2vec、上下文相关embedding ELMo、BERT、字符embedding基于CNN或者LSTM的CharEmbedding

与此同时fastNLP 在 modules 模块中内置了两种模块的诸多组件,可以帮助用户快速搭建自己所需的网络。 两种模块的功能和常见组件如下:

类型 功能 例子
encoder 将输入编码为具有具有表示能力的向量 embedding, RNN, CNN, transformer
decoder 将具有某种表示意义的向量解码为需要的输出形式 MLP, CRF

项目结构

fastNLP的大致工作流程如上图所示而项目结构如下

fastNLP 开源的自然语言处理库
fastNLP.core 实现了核心功能,包括数据处理组件、训练器、测试器等
fastNLP.models 实现了一些完整的神经网络模型
fastNLP.modules 实现了用于搭建神经网络模型的诸多组件
fastNLP.embeddings 实现了将序列index转为向量序列的功能包括读取预训练embedding等
fastNLP.io 实现了读写功能,包括数据读入,模型读写等

In memory of @FengZiYjun. May his soul rest in peace. We will miss you very very much!