TensorFlow Hub 是一个用于共享可重用机器学习模型组件的库
pip install tensorflow tensorflow_hub
import tensorflow as tf
import tensorflow_hub as hub
module_url = "https://tfhub.dev/google/tf2-preview/gnews-swivel-20dim/1"
hub.load()
函数加载模块:embed = hub.load(module_url)
"default"
:input_text = ["Hello world!", "How are you?"]
embeddings = embed(input_text)
print(embeddings)
这将输出一个形状为 (len(input_text), embedding_dim)
的张量,其中 embedding_dim
是嵌入向量的维度。
这是一个完整的示例:
import tensorflow as tf
import tensorflow_hub as hub
module_url = "https://tfhub.dev/google/tatf2-preview/gnews-swivel-20dim/1"
embed = hub.load(module_url)
input_text = ["Hello world!", "How are you?"]
embeddings = embed(input_text)
print(embeddings)
领取专属 10元无门槛券
手把手带您无忧上云