TensorFlow Hub 是一種程式庫,內容是可重複利用的機器學習模組。

  !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 程式庫是用來發布、探索和使用機器學習模型中可重複利用的部分。模組是 TensorFlow 圖形的獨立部分,包含權重和資產,可在不同的工作中重複使用,這個流程稱為「遷移學習」。遷移學習可以:
  • 使用較小的資料集訓練模型
  • 改善一般化功能
  • 加快訓練速度