TensorFlow Hub est un référentiel ouvert et une bibliothèque pour l'apprentissage automatique réutilisable. Le référentiel hub.tensorflow.google.cn fournit de nombreux modèles pré-formé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://hub.tensorflow.google.cn/google/nnlm-en-dim128/2")
embeddings = model(["The rain in Spain.", "falls",
"mainly", "In the plain!"])
print(embeddings.shape) #(4,128)