KerasHub Modeling API
The following base classes form the API for working with pretrained models
through KerasHub. These base classes can be used with the from_preset()
constructor to automatically instantiate a subclass with the correct model
architecture, e.g.
keras_hub.models.TextClassifier.from_preset("bert_base_en", num_classes=2).
For the full list of available pretrained model presets shipped directly by the Keras team, see the Pretrained Models page.
Backbone
- Backbone class
- from_preset method
- token_embedding property
- enable_lora method
- save_lora_weights method
- load_lora_weights method
- save_to_preset method
CausalLM
- CausalLM class
- from_preset method
- compile method
- generate method
- save_to_preset method
- preprocessor property
- backbone property
CausalLMPreprocessor
ImageClassifier
- ImageClassifier class
- from_preset method
- compile method
- save_to_preset method
- preprocessor property
- backbone property
ImageClassifierPreprocessor
ImageToImage
- ImageToImage class
- from_preset method
- compile method
- save_to_preset method
- preprocessor property
- backbone property
- generate method
ImageSegmenter
- ImageSegmenter class
- from_preset method
- compile method
- save_to_preset method
- preprocessor property
- backbone property
Inpaint
- Inpaint class
- from_preset method
- compile method
- save_to_preset method
- preprocessor property
- backbone property
- generate method
MaskedLM
- MaskedLM class
- from_preset method
- compile method
- save_to_preset method
- preprocessor property
- backbone property
MaskedLMPreprocessor
Preprocessor
Seq2SeqLM
- Seq2SeqLM class
- from_preset method
- compile method
- generate method
- save_to_preset method
- preprocessor property
- backbone property
Seq2SeqLMPreprocessor
Task
TextClassifier
- TextClassifier class
- from_preset method
- compile method
- save_to_preset method
- preprocessor property
- backbone property
TextClassifierPreprocessor
TextToImage
- TextToImage class
- from_preset method
- compile method
- save_to_preset method
- preprocessor property
- backbone property
- generate method