TensorFlow Hub es un repositorio de modelos de aprendizaje automático capacitados.
!pip install --upgrade tensorflow_hub
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)
TensorFlow Hub es un repositorio de modelos de aprendizaje automático capacitados, listos para ajustarse e implementarse en cualquier lugar. Reutilice modelos entrenados como BERT y Faster R-CNN con solo unas pocas líneas de código.
-
ver la guía
Obtenga información sobre cómo usar TensorFlow Hub y cómo funciona. -
Ver tutoriales
Los tutoriales le muestran ejemplos completos con TensorFlow Hub. -
ver modelos
Encuentre modelos TF, TFLite y TF.js entrenados para su caso de uso.
Modelos
Encuentre modelos capacitados de la comunidad de TensorFlow en TFHub.dev
BERT
Consulte BERT para tareas de PNL, incluida la clasificación de texto y la respuesta a preguntas.
Detección de objetos,Detección de objetos
Utilice el modelo Faster R-CNN Inception ResNet V2 640x640 para detectar objetos en imágenes.
Transferencia de estilo
Transfiere el estilo de una imagen a otra utilizando el modelo de transferencia de estilo de imagen.
Clasificador de alimentos en el dispositivo
Utilice este modelo TFLite para clasificar fotos de alimentos en un dispositivo móvil.
Noticias y anuncios
Visite nuestro blog para ver más anuncios y vea las últimas actualizaciones de #TFHub en Twitter
TensorFlow Hub para impacto en el mundo real en Google I/O
Aprende cómo puedes usar TensorFlow Hub para crear soluciones de ML con impacto en el mundo real.
Soluciones de aprendizaje automático en el dispositivo
Para explorar las soluciones de ML para sus aplicaciones móviles y web, incluido TensorFlow Hub, visite la página de aprendizaje automático en el dispositivo de Google.
Simplificar BERT con modelos de preprocesamiento de TensorFlow Hub
TensorFlow Hub hace que BERT sea fácil de usar con nuevos modelos de preprocesamiento.
Del canto a las partituras musicales: estimación del tono con SPICE y Tensorflow Hub
Aprenda a usar el modelo SPICE para transcribir automáticamente partituras de audio en vivo.
Comunidad
Únase a la comunidad de TensorFlow Hub
