KerasNLP
在 TensorFlow 中开始处理文本的最简单方法是使用 KerasNLP,这是一个自然语言处理库,它提供具有最先进的预设权重和架构的模块化组件。您可以开箱即用地使用 KerasNLP 组件,也可以根据需要对其进行自定义。KerasNLP 强调所有工作流程中的图内计算,因此您可以期待使用 TensorFlow 生态系统轻松地进行生产化。
要安装 KerasNLP,请参阅 安装。
TensorFlow 文本
tensorflow_text
包提供了一组与文本相关的类和操作,可以与 TensorFlow 一起使用。该库可以执行文本模型通常需要的预处理,并包含其他对核心 TensorFlow 未提供的序列建模有用的功能。
有关安装详细信息,请参阅 指南
TensorFlow 模型 - NLP
TensorFlow 模型存储库 提供了最先进 (SOTA) 模型的实现。 tensorflow-models-official
pip 包包含许多用于构建 SOTA NLP 模型的高级函数和类,包括 nlp.layers
、nlp.losses
、nlp.models
和 nlp.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