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
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 | Sí | Sí |
PyTorch, TensorFlow |
|
Detección de objetos | Sí | Sí | PyTorch, TensorFlow, MXNet | |
Segmentación semántica | Sí | Sí | MXNet | |
Segmentación de instancias | Sí | Sí | MXNet | |
Incrustación de imágenes | Sí | No | TensorFlow, MXNet | |
Clasificación de textos | Sí | Sí | TensorFlow | |
Clasificación de pares de frases | Sí | Sí | TensorFlow, Hugging Face | |
Respuesta a preguntas | Sí | Sí | PyTorch, Hugging Face | |
Reconocimiento de entidades con nombre | Sí | No | Hugging Face |
Introducción a JumpStart: reconocimiento de entidades con nombre |
Resumen de texto | Sí | No | Hugging Face | |
Generación de texto | Sí | No | Hugging Face | |
Traducción automática | Sí | No | Hugging Face | |
Incrustación de texto | Sí | No | TensorFlow, MXNet | |
Clasificación tabular | Sí | Sí | 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 | Sí | Sí | 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 |