Cette page a été traduite par l'API Cloud Translation.
Switch to English

Hub TensorFlow

TensorFlow Hub est un référentiel et une bibliothèque ouverts pour l'apprentissage automatique réutilisable. Le référentiel tfhub.dev fournit de nombreux modèles pré-entraînés: incorporations de texte, modèles de classification d'images, modèles TF.js / TFLite et bien plus encore. Le référentiel est ouvert aux contributeurs de la communauté .

La bibliothèque tensorflow_hub vous permet de les télécharger et de les réutiliser dans votre programme TensorFlow avec un minimum de code.

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)

Prochaines étapes