TensorFlow Hub est une bibliothèque de modules de machine learning réutilisables.

  !pip install "tensorflow_hub>=0.6.0"
  !pip install "tensorflow>=2.0.0"

  import tensorflow as tf
  import tensorflow_hub as hub

  module_url = "https://tfhub.dev/google/nnlm-en-dim128/2"
  embed = hub.KerasLayer(module_url)
  embeddings = embed(["A long sentence.", "single-word",
                      "http://example.com"])
  print(embeddings.shape)  #(3,128)
TensorFlow Hub est une bibliothèque conçue pour favoriser la publication, la découverte et l'exploitation des parties réutilisables de modèles de machine learning. Un module est un élément autonome d'un graphe TensorFlow, comprenant ses pondérations et ses ressources, qui peut être réutilisé pour différentes tâches d'un processus connu sous le nom d'apprentissage par transfert. L'apprentissage par transfert permet :
  • d'entraîner un modèle avec un ensemble de données plus petit,
  • d'améliorer la généralisation,
  • d'accélérer l'entraînement.