TensorFlow Hub

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)

次のステップ