以下基礎類別組成了透過 KerasHub 使用預訓練模型的 API。這些基礎類別可以與 from_preset() 建構子一起使用,以自動實例化具有正確模型架構的子類別,例如 keras_hub.models.TextClassifier.from_preset("bert_base_en", num_classes=2)。
from_preset()
keras_hub.models.TextClassifier.from_preset("bert_base_en", num_classes=2)
有關 Keras 團隊直接提供的完整可用預訓練模型預設列表,請參閱預訓練模型頁面。