TensorFlow Hub 程式庫總覽

tensorflow_hub 程式庫可讓您在 TensorFlow 程式中下載並重複使用已訓練的模型,且只需少量的程式碼。hub.KerasLayer API 是載入已訓練模型的主要方式。

import tensorflow_hub as hub

embed = hub.KerasLayer("https://tfhub.dev/google/nnlm-en-dim128/2")
embeddings = embed(["A long sentence.", "single-word", "http://example.com"])
print(embeddings.shape, embeddings.dtype)

注意:本文件範例中使用 TFhub.dev 網址控制代碼。如要進一步瞭解其他有效的控制代碼類型,請參閱這裡

設定下載的快取位置。

根據預設,tensorflow_hub 會使用全系統的暫時目錄來快取已下載和未壓縮的模型。如要瞭解使用其他可能更持久位置的選項,請參閱快取

API 穩定性

雖然我們希望防止重大變更,但這個專案仍在積極開發中,目前尚不保證具有穩定的 API 或模型格式。

公平性

如同所有機器學習一樣,公平性是一個重要的考量因素。許多預先訓練的模型都是在大型資料集上訓練而成。重複使用任何模型時,務必留意模型的訓練資料 (以及其中是否有任何既有偏見),以及這些資料可能對您的使用方式造成的影響。

安全性

由於模型包含任意 TensorFlow 圖表,因此可以將模型視為程式。安全地使用 TensorFlow 說明從不受信任來源參照模型的安全性意涵。

後續步驟