Realice un ajuste automático del modelo con SageMaker - Amazon SageMaker

Las traducciones son generadas a través de traducción automática. En caso de conflicto entre la traducción y la version original de inglés, prevalecerá la version en inglés.

Realice un ajuste automático del modelo con SageMaker

El ajuste SageMaker automático de modelos (AMT) de Amazon busca la mejor versión de un modelo realizando muchos trabajos de entrenamiento en su conjunto de datos. El ajuste SageMaker automático de modelos (AMT) de Amazon también se conoce como ajuste de hiperparámetros. Para ello, AMT utiliza el algoritmo y los rangos de hiperparámetros que especifique. A continuación, elige los valores de hiperparámetros que dan lugar a un modelo con el mejor rendimiento medido por una métrica de su elección.

Por ejemplo, resolver un problema de clasificación binaria en un conjunto de datos de marketing. Su objetivo es maximizar la métrica área bajo la curva (AUC) del algoritmo formando un modelo Usa el algoritmo XGBoost con Amazon SageMaker. Desea encontrar los valores de los hiperparámetros eta, alpha, min_child_weight y max_depth que servirán para formar el mejor modelo. Especifique un rango de valores para estos hiperparámetros. A continuación, el ajuste de SageMaker hiperparámetros busca dentro de los rangos para encontrar una combinación que cree un trabajo de formación que cree un modelo con el AUC más alto. Para conservar los recursos o cumplir con una expectativa de calidad específica del modelo, configure los criterios de finalización para detener el ajuste una vez que se hayan cumplido los criterios.

Puede usar SageMaker AMT con algoritmos integrados, algoritmos personalizados o contenedores SageMaker prediseñados para los marcos de aprendizaje automático.

SageMaker AMT puede utilizar una instancia de Amazon EC2 Spot para optimizar los costes al ejecutar tareas de formación. Para obtener más información, consulte Utilice Managed Spot Training en Amazon SageMaker.

Antes de comenzar a utilizar el ajuste de hiperparámetros, debe tener un problema de machine learning bien definido que incluya los siguientes elementos:

  • Un conjunto de datos

  • La comprensión del tipo de algoritmo que necesita formar

  • Una comprensión clara de cómo medir el éxito

Prepare el conjunto de datos y el algoritmo para que funcionen SageMaker y ejecuten correctamente un trabajo de formación al menos una vez. Para obtener información acerca de cómo configurar y ejecutar un trabajo de entrenamiento, consulte Guía para configurar Amazon SageMaker.