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

KerasHub API 文档

KerasHub 是一个模块化构建块的工具箱,从预训练的最先进模型到低级别的 Transformer Encoder 层应有尽有。

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

建模 API

模型架构

分词器

预处理层

建模层

采样器

指标