Modelos específicos para tareas - Amazon SageMaker

Modelos específicos para tareas

JumpStart admite modelos de tareas específicas para quince de los tipos de problemas más populares. De los tipos de problemas admitidos, los relacionados con la visión y el NLP suman un total de trece. Hay ocho tipos de problemas que permiten el entrenamiento y el ajuste incrementales. Para obtener más información sobre el entrenamiento incremental y el ajuste de hiperparámetros, consulte Ajuste automático de modelos de SageMaker. JumpStart también admite cuatro algoritmos populares para el modelado de datos tabulares.

Puede buscar y explorar modelos desde la página de inicio de JumpStart en Studio o Studio Classic. Al seleccionar un modelo, la página de detalles del modelo proporciona información sobre el modelo, y puede entrenarlo e implementarlo en unos pocos pasos. La sección de descripción describe lo que puede hacer con el modelo, los tipos esperados de entradas y salidas y el tipo de datos necesarios para ajustar el modelo.

También puede utilizar modelos mediante programación con el SageMaker Python SDK. Para obtener una lista de todos los modelos disponibles, consulte la Tabla de modelos disponibles de JumpStart.

En la siguiente tabla se resume la lista de tipos de problemas y los enlaces a sus ejemplos de cuadernos de Jupyter.

Tipos de problemas Compatibilidad con la inferencia con modelos previamente entrenados Entrenable en un conjunto de datos personalizado Marcos admitidos Cuadernos de ejemplo
Clasificación de imágenes

PyTorch, TensorFlow

Introducción a JumpStart: clasificación de imágenes

Detección de objetos PyTorch, TensorFlow, MXNet

Introducción a JumpStart: detección de objetos

Segmentación semántica MXNet

Introducción a JumpStart: segmentación semántica

Segmentación de instancias MXNet

Introducción a JumpStart: segmentación de instancias

Incrustación de imágenes No TensorFlow, MXNet

Introducción a JumpStart: incrustación de imágenes

Clasificación de textos TensorFlow

Introducción a JumpStart: clasificación de texto

Clasificación de pares de frases TensorFlow, Hugging Face

Introducción a JumpStart: clasificación de pares de frases

Respuesta a preguntas PyTorch, Hugging Face

Introducción a JumpStart: respuesta a preguntas

Reconocimiento de entidades con nombre No Hugging Face

Introducción a JumpStart: reconocimiento de entidades con nombre

Resumen de texto No Hugging Face

Introducción a JumpStart: resumen de texto

Generación de texto No Hugging Face

Introducción a JumpStart: generación de texto

Traducción automática No Hugging Face

Introducción a JumpStart: traducción automática

Incrustación de texto No TensorFlow, MXNet

Introducción a JumpStart: incrustación de imágenes

Clasificación tabular LightGBM, CatBoost, XGBoost, AutoGluon-Tabular, TabTransformer, aprendizaje lineal

Introducción a JumpStart - Clasificación tabular - LightGBM, CatBoost

Introducción a JumpStart - Clasificación tabular - XGBoost, aprendizaje lineal

Introducción a JumpStart - Clasificación tabular - AutoGluon Learner

Introducción a JumpStart - Clasificación tabular - TabTransformer Learner

Regresión tabular LightGBM, CatBoost, XGBoost, AutoGluon-Tabular, TabTransformer, aprendizaje lineal

Introducción a JumpStart - Regresión tabular - LightGBM, CatBoost

Introducción a JumpStart - Regresión tabular - XGBoost, aprendizaje lineal

Introducción a JumpStart - Regresión tabular - AutoGluon, aprendizaje lineal

Introducción a JumpStart - Regresión tabular - TabTransformer Learner