文本和 NLP 库的 API 参考。

KerasNLP

API 参考

在 TensorFlow 中开始处理文本的最简单方法是使用 KerasNLP,这是一个自然语言处理库,它提供具有最先进的预设权重和架构的模块化组件。您可以开箱即用地使用 KerasNLP 组件,也可以根据需要对其进行自定义。KerasNLP 强调所有工作流程中的图内计算,因此您可以期待使用 TensorFlow 生态系统轻松地进行生产化。

要安装 KerasNLP,请参阅 安装

TensorFlow 文本

API 参考

tensorflow_text 包提供了一组与文本相关的类和操作,可以与 TensorFlow 一起使用。该库可以执行文本模型通常需要的预处理,并包含其他对核心 TensorFlow 未提供的序列建模有用的功能。

有关安装详细信息,请参阅 指南

TensorFlow 模型 - NLP

API 参考

TensorFlow 模型存储库 提供了最先进 (SOTA) 模型的实现。 tensorflow-models-official pip 包包含许多用于构建 SOTA NLP 模型的高级函数和类,包括 nlp.layersnlp.lossesnlp.modelsnlp.tasks

您可以使用 pip 安装该包

$ pip install tensorflow-models-official  # For the latest release
$ #or
$ pip install tf-models-nightly # For the nightly build

NLP 功能在 tfm.nlp 子模块中可用。

import tensorflow_models as tfm
tfm.nlp