要使用TensorFlow Hub,您需要首先安裝TensorFlow和TensorFlow Hub庫。然后,您可以使用TensorFlow Hub庫中提供的預訓練模型和特征提取器來進行遷移學習或直接使用這些模型進行預測。
以下是TensorFlow Hub的基本使用方法:
import tensorflow as tf
import tensorflow_hub as hub
module = hub.load("https://tfhub.dev/google/universal-sentence-encoder/4")
embeddings = module(["Hello world", "TensorFlow is awesome"])
print(embeddings)
base_model = hub.KerasLayer("https://tfhub.dev/google/imagenet/mobilenet_v2_130_224/classification/4")
model = tf.keras.Sequential([
base_model,
tf.keras.layers.Dense(10, activation='softmax')
])
通過這些步驟,您可以輕松地使用TensorFlow Hub來利用預訓練模型進行特征提取、遷移學習或直接使用預訓練模型進行預測。