TensorFlow Hub es un repositorio de modelos de aprendizaje automático entrenados.
!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 usando 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 que incluyen clasificación de texto y respuesta a preguntas.
Detección de objetos
Utilice el modelo Faster R-CNN Inception ResNet V2 640x640 para detectar objetos en imágenes.
Transferencia de estilo
Transfiera el estilo de una imagen a otra usando el modelo de transferencia de estilo de imagen.
Clasificador de alimentos en el dispositivo
Utilice este modelo TFLite para clasificar fotografías de comida 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 un impacto en el mundo real en Google I/O
Descubra cómo puede utilizar TensorFlow Hub para crear soluciones de aprendizaje automático con impacto en el mundo real.
Soluciones de aprendizaje automático en el dispositivo
Para explorar soluciones de aprendizaje automático para sus aplicaciones web y móviles, incluido TensorFlow Hub, visite la página de aprendizaje automático en el dispositivo de Google.
Hacer BERT más fácil 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 utilizar el modelo SPICE para transcribir automáticamente partituras a partir de audio en vivo.
Comunidad
Únase a la comunidad de TensorFlow Hub