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

KerasHub 模型 API

以下基类构成了通过 KerasHub 使用预训练模型的 API。这些基类可以与 from_preset() 构造函数一起使用,以自动实例化具有正确模型架构的子类,例如 keras_hub.models.TextClassifier.from_preset("bert_base_en", num_classes=2)

有关 Keras 团队直接提供的可用预训练模型预设的完整列表,请参见预训练模型页面。

主干

因果语言模型 (CausalLM)

因果语言模型预处理器 (CausalLMPreprocessor)

图像分类器 (ImageClassifier)

图像分类器预处理器 (ImageClassifierPreprocessor)

图像到图像 (ImageToImage)

图像分割器 (ImageSegmenter)

图像修复 (Inpaint)

掩码语言模型 (MaskedLM)

掩码语言模型预处理器 (MaskedLMPreprocessor)

预处理器 (Preprocessor)

序列到序列语言模型 (Seq2SeqLM)

序列到序列语言模型预处理器 (Seq2SeqLMPreprocessor)

任务 (Task)

文本分类器 (TextClassifier)

文本分类器预处理器 (TextClassifierPreprocessor)

文本到图像 (TextToImage)

上传预设 (upload_preset)