TensorFlow Hub
Restez organisé à l'aide des collections
Enregistrez et classez les contenus selon vos préférences.
TensorFlow Hub est un référentiel ouvert et une bibliothèque pour l'apprentissage automatique réutilisable. Le référentiel tfhub.dev fournit de nombreux modèles pré-entraînés : intégrations de texte, modèles de classification d'images, modèles TF.js/TFLite et bien plus encore. Le référentiel est ouvert aux contributeurs de la communauté .
La bibliothèque tensorflow_hub
vous permet de les télécharger et de les réutiliser dans votre programme TensorFlow avec un minimum de code.
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)
Prochaines étapes
Sauf indication contraire, le contenu de cette page est régi par une licence Creative Commons Attribution 4.0, et les échantillons de code sont régis par une licence Apache 2.0. Pour en savoir plus, consultez les Règles du site Google Developers. Java est une marque déposée d'Oracle et/ou de ses sociétés affiliées.
Dernière mise à jour le 2025/07/25 (UTC).
[null,null,["Dernière mise à jour le 2025/07/25 (UTC)."],[],[],null,["# TensorFlow Hub\n\n\u003cbr /\u003e\n\nTensorFlow Hub is an open repository and library for reusable machine learning.\nThe [tfhub.dev](https://tfhub.dev) repository provides many pre-trained models:\ntext embeddings, image classification models, TF.js/TFLite models and much more.\nThe repository is open to\n[community contributors](https://tfhub.dev/s?subtype=publisher).\n\nThe [`tensorflow_hub`](https://github.com/tensorflow/hub) library lets you\ndownload and reuse them in your TensorFlow program with a minimum amount of\ncode. \n\n import tensorflow_hub as hub\n\n model = hub.KerasLayer(\"https://tfhub.dev/google/nnlm-en-dim128/2\")\n embeddings = model([\"The rain in Spain.\", \"falls\",\n \"mainly\", \"In the plain!\"])\n\n print(embeddings.shape) #(4,128)\n\nNext Steps\n----------\n\n- [Find models on tfhub.dev](https://tfhub.dev)\n- [Publish models on tfhub.dev](/hub/publish)\n- TensorFlow Hub library\n - [Install TensorFlow Hub](/hub/installation)\n - [Library overview](/hub/lib_overview)\n- [Follow tutorials](/hub/tutorials)"]]