以下基类构成了通过 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 团队直接提供的可用预训练模型预设的完整列表,请参见预训练模型页面。