KerasHub 建模 API

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

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

Backbone

• Backbone 类
• from_preset 方法
• token_embedding 属性
• enable_lora 方法
• save_lora_weights 方法
• load_lora_weights 方法
• save_to_preset 方法

CausalLM

• CausalLM 类
• from_preset 方法
• compile 方法
• generate 方法
• save_to_preset 方法
• preprocessor 属性
• backbone 属性

CausalLMPreprocessor

• CausalLMPreprocessor 类
• from_preset 方法
• save_to_preset 方法
• tokenizer 属性

ImageClassifier

• ImageClassifier 类
• from_preset 方法
• compile 方法
• save_to_preset 方法
• preprocessor 属性
• backbone 属性

ImageClassifierPreprocessor

• ImageClassifierPreprocessor 类
• from_preset 方法
• save_to_preset 方法

ImageToImage

• ImageToImage 类
• from_preset 方法
• compile 方法
• save_to_preset 方法
• preprocessor 属性
• backbone 属性
• generate 方法

ImageSegmenter

• ImageSegmenter 类
• from_preset 方法
• compile 方法
• save_to_preset 方法
• preprocessor 属性
• backbone 属性

Inpaint

• Inpaint 类
• from_preset 方法
• compile 方法
• save_to_preset 方法
• preprocessor 属性
• backbone 属性
• generate 方法

MaskedLM

• MaskedLM 类
• from_preset 方法
• compile 方法
• save_to_preset 方法
• preprocessor 属性
• backbone 属性

MaskedLMPreprocessor

• MaskedLMPreprocessor 类
• from_preset 方法
• save_to_preset 方法
• tokenizer 属性

Preprocessor

• Preprocessor 类
• from_preset 方法
• save_to_preset 方法
• tokenizer 属性

Seq2SeqLM

• Seq2SeqLM 类
• from_preset 方法
• compile 方法
• generate 方法
• save_to_preset 方法
• preprocessor 属性
• backbone 属性

Seq2SeqLMPreprocessor

• Seq2SeqLMPreprocessor 类
• from_preset 方法
• save_to_preset 方法
• tokenizer 属性

Task

• Task 类
• from_preset 方法
• save_to_preset 方法
• preprocessor 属性
• backbone 属性

TextClassifier

• TextClassifier 类
• from_preset 方法
• compile 方法
• save_to_preset 方法
• preprocessor 属性
• backbone 属性

TextClassifierPreprocessor

• TextClassifierPreprocessor 类
• from_preset 方法
• save_to_preset 方法
• tokenizer 属性

TextToImage

• TextToImage 类
• from_preset 方法
• compile 方法
• save_to_preset 方法
• preprocessor 属性
• backbone 属性
• generate 方法

upload_preset

• upload_preset 函数