Halaman ini diterjemahkan oleh Cloud Translation API.
Switch to English

TensorFlow Hub

TensorFlow Hub adalah repositori dan library terbuka untuk machine learning yang dapat digunakan kembali. Repositori tfhub.dev menyediakan banyak model terlatih: embeddings teks, model klasifikasi gambar, model TF.js / TFLite, dan banyak lagi. Repositori terbuka untuk kontributor komunitas .

Pustaka tensorflow_hub memungkinkan Anda mendownload dan menggunakannya kembali dalam program TensorFlow dengan jumlah kode minimum.

import tensorflow_hub as hub

model = hub.KerasLayer("https://tfhub.dev/google/nnlm-en-dim128/2")
embeddings = model(["The rain in Spain.", "falls",
                    "mainly", "In the plain!"])

print(embeddings.shape)  #(4,128)

Langkah selanjutnya