適用於文字和 NLP 程式庫的 API 參考文件。

KerasNLP

API 參考文件

在 TensorFlow 中開始處理文字最簡單的方式是使用 KerasNLP,這是一個自然語言處理程式庫,提供模組化元件以及最先進的預設權重和架構。您可以直接使用 KerasNLP 元件,或根據需要自訂。KerasNLP 強調所有工作流程的圖內運算,因此您可以期望使用 TensorFlow 生態系統輕鬆實現生產環境部署。

若要安裝 KerasNLP,請參閱安裝

TensorFlow Text

API 參考文件

tensorflow_text 套件提供了一系列與文字相關的類別和運算,可隨時與 TensorFlow 搭配使用。此程式庫可以執行以文字為基礎的模型經常需要的預先處理,並包含核心 TensorFlow 未提供的其他序列模型功能。

如需安裝詳細資訊,請參閱指南

TensorFlow Models - NLP

API 參考文件

TensorFlow Models 存放區提供最先進 (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