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)