Menggunakan Keras di Hugging Face

Keras adalah framework deep learning multi-backend open-source, dengan dukungan untuk JAX, TensorFlow, dan PyTorch. Anda dapat menemukan detail lebih lanjut tentangnya di keras.io.

Menjelajahi Keras di Hub

Anda dapat menampilkan model keras di Hub dengan memfilter berdasarkan nama pustaka di halaman model.

Model Keras di Hub memiliki fitur yang berguna saat diunggah langsung dari pustaka Keras:

1. Kartu model yang dihasilkan dengan deskripsi, plot model, dan lainnya.
2. Jumlah unduhan untuk memantau popularitas model.
3. Potongan kode untuk memulai dengan model secara cepat.

Menggunakan model yang sudah ada

Keras terintegrasi secara mendalam dengan Hugging Face Hub. Artinya, Anda dapat memuat dan menyimpan model di Hub langsung dari pustaka. Untuk melakukannya, Anda perlu menginstal versi terbaru Keras dan huggingface_hub. Pustaka huggingface_hub adalah klien Python ringan yang digunakan oleh Keras untuk berinteraksi dengan Hub.

pip install -U keras huggingface_hub

Setelah pustaka terpasang, Anda hanya perlu menggunakan metode keras.saving.load_model reguler dengan meneruskan jalur Hugging Face sebagai argumen. Jalur HF adalah repo_id yang diawali dengan hf://, misalnya "hf://keras-io/weather-prediction". Baca lebih lanjut tentang load_model di dokumentasi Keras.

import keras

model = keras.saving.load_model("hf://Wauplin/mnist_example")

Jika Anda ingin melihat cara memuat model tertentu, Anda dapat mengklik Gunakan model ini di halaman model untuk mendapatkan potongan kode yang berfungsi!

Berbagi model Anda

Sama seperti load_model, Anda dapat menyimpan dan membagikan model keras di Hub menggunakan model.save() dengan path HF:

model = ...
model.save("hf://your-username/your-model-name")

Jika repositori tidak ada di Hub, repositori tersebut akan dibuat untuk Anda. Model yang diunggah berisi kartu model, plot model, file metadata.json dan config.json, serta file model.weights.h5 yang berisi bobot model.

Secara default, repositori akan berisi kartu model minimal. Lihat panduan Kartu Model untuk mempelajari lebih lanjut tentang kartu model dan cara melengkapinya. Anda juga dapat memperbarui kartu model secara programatik menggunakan huggingface_hub.ModelCard (lihat panduan).

Anda mungkin sudah familiar dengan file .keras. Sebenarnya, file .keras hanyalah file zip yang berisi file .json dan model.weights.h5. Saat diunggah ke Hub, model disimpan sebagai folder yang sudah diekstrak agar Anda dapat menavigasi file-file tersebut. Perhatikan bahwa jika Anda mengunggah file .keras secara manual ke repositori model di Hub, repositori tersebut akan secara otomatis diberi tag keras tetapi Anda tidak akan dapat memuatnya menggunakan keras.saving.load_model.

Sumber daya tambahan

• Panduan Developer Keras.
• Contoh Keras.