ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

TensorFlow Hub

TensorFlow Hub is an open repository and library for reusable machine learning. The repository provides many pre-trained models: text embeddings, image classification models, TF.js/TFLite models and much more. The repository is open to community contributors.

The tensorflow_hub library lets you download and reuse them in your TensorFlow program with a minimum amount of code.

import tensorflow_hub as hub

model = hub.KerasLayer("")
embeddings = model(["The rain in Spain.", "falls",
                    "mainly", "In the plain!"])

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

Next Steps