TF-Keras（舊版）

Hub

tf-keras 是 Keras 2.x 版本的名稱。它現在作為獨立的 GitHub 倉庫在此託管。儘管它是一箇舊版框架，但 Hub 上仍有4.5k+ 個模型。這些模型可以使用 huggingface_hub 庫載入。您**必須**在您的機器上安裝 tf-keras 或 keras<3.x。

如果您對 Keras 3.x 支援感興趣，請檢視此指南。

安裝後，您只需使用 from_pretrained_keras 方法即可從 Hub 載入模型。有關 from_pretrained_keras 的更多資訊，請在此閱讀。

from huggingface_hub import from_pretrained_keras

model = from_pretrained_keras("keras-io/mobile-vit-xxs")
prediction = model.predict(image)
prediction = tf.squeeze(tf.round(prediction))
print(f'The image is a {classes[(np.argmax(prediction))]}!')

# The image is a sunflower!

您也可以將 tf-keras 模型託管在 Hub 上。但是，請記住 tf-keras 是一箇舊版框架。為了覆蓋最多的使用者，我們建議使用 Keras 3.x 建立模型並如上所述原生共享。有關上傳 tf-keras 模型的更多詳細資訊，請檢視push_to_hub_keras 文件。

from huggingface_hub import push_to_hub_keras

push_to_hub_keras(model,
    "your-username/your-model-name",
    "your-tensorboard-log-directory",
    tags = ["object-detection", "some_other_tag"],
    **model_save_kwargs,
)

其他資源

GitHub 倉庫

部落格文章將 Keras 放入 🤗 Hub 進行協作訓練和可復現性（2022 年 4 月）

< > 在 GitHub 上更新