Pusat TensorFlow

Tetap teratur dengan koleksi Simpan dan kategorikan konten berdasarkan preferensi Anda.

TensorFlow Hub adalah repositori dan library terbuka untuk machine learning yang dapat digunakan kembali. The tfhub.dev repositori menyediakan banyak pra-dilatih model: embeddings teks, model klasifikasi citra, TF.js / model TFLite dan banyak lagi. Repositori ini terbuka untuk kontributor masyarakat .

The tensorflow_hub perpustakaan memungkinkan Anda men-download dan menggunakan kembali mereka dalam program TensorFlow Anda dengan jumlah minimum kode.

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