このページは Cloud Translation API によって翻訳されました。
Switch to English

TensorFlowハブ

TensorFlow Hubは、再利用可能な機械学習のためのオープンリポジトリおよびライブラリです。 tfhub.devリポジトリは、テキスト埋め込み、画像分類モデル、TF.js / TFLiteモデルなどの多くの事前トレーニング済みモデルを提供します。リポジトリはコミュニティの貢献者に公開されています

tensorflow_hubライブラリを使用すると、最小限のコードでそれらをダウンロードしてTensorFlowプログラムで再利用できます。

import tensorflow_hub as hub

model = hub.KerasLayer("https://tfhub.dev/google/nnlm-en-dim128/2")
embeddings = model(["The rain in Spain.", "falls",
                    "mainly", "In the plain!"])

print(embeddings.shape)  #(4,128)

次のステップ