O TensorFlow Hub é uma biblioteca de módulos de machine learning reutilizáveis.

  !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)
O TensorFlow Hub é uma biblioteca criada para a publicação, a descoberta e o consumo de partes reutilizáveis de modelos de machine learning. Um módulo é uma parte autônoma de um grafo do TensorFlow, com os respectivos pesos e recursos, que podem ser reutilizados em diferentes tarefas em um processo conhecido como aprendizado por transferência. O aprendizado de transferência pode:
  • treinar um modelo com um conjunto de dados menor;
  • melhorar a generalização;
  • acelerar o treinamento.