TensorFlow Hub est un référentiel ouvert et une bibliothèque pour l'apprentissage automatique réutilisable. Le référentiel tfhub.dev fournit de nombreux modèles pré-entraînés : intégrations 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)