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

KerasHub 建模 API

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

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

Backbone

CausalLM

CausalLMPreprocessor

ImageClassifier

ImageClassifierPreprocessor

ImageToImage

ImageSegmenter

Inpaint

MaskedLM

MaskedLMPreprocessor

Preprocessor

Seq2SeqLM

Seq2SeqLMPreprocessor

Task

TextClassifier

TextClassifierPreprocessor

TextToImage

upload_preset