KerasHub:预训练模型 / KerasHub API 文档

KerasHub API 文档

KerasHub 是一个模块化构建块工具箱,范围涵盖最先进的预训练模型到低级的 Transformer Encoder 层。

  • 建模 API:可用于大多数使用预训练模型的高级任务的基础类。请注意,您可以使用基础类上的 from_preset() 构造函数来实例化正确子类的模型。
  • 模型架构:KerasHub 中随附的所有预训练模型架构的实现。
  • 分词器:用于基于文本模型的标记化例程的层实现。
  • 预处理层:用于构建处理音频、文本和图像输入的预处理管道的层。
  • 建模层:预训练模型架构使用的常用建模层。
  • 采样器:用于控制生成式文本采样的 API。
  • 指标:对音频、文本和图像工作流有用的指标。

建模 API

模型架构

分词器

预处理层

建模层

采样器

指标