Ajustar un modelo DeepAR - 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.

Ajustar un modelo DeepAR

El ajuste de modelo automático, también conocido como ajuste de hiperparámetros, encuentra la mejor versión de un modelo ejecutando muchas tareas que probar una serie de hiperparámetros en su conjunto de datos. Usted elige los hiperparámetros que pueden ajustarse, un rango de valores para cada uno de ellos y una métrica objetiva. Puede elegir la métrica objetiva de las métricas que el algoritmo computa. El ajuste de modelo automático busca los hiperparámetros elegidos para encontrar la combinación de valores que obtienen el modelo que optimiza la métrica objetiva.

Para obtener más información acerca del ajuste de modelos, consulte Ajuste automático del modelo con SageMaker.

Métricas calculadas por el algoritmo DeepAR

El algoritmo DeepAR notifica tres métricas, que se calculan durante la capacitación. Al ajustar un modelo, elija una de opciones como objetivo. Para el objetivo, utilice la precisión de previsión en un canal de prueba proporcionado (recomendado) o la pérdida de capacitación. Para recomendaciones para la división de capacitación/de pruebas para el algoritmo DeepAR, consulte Prácticas recomendadas para utilizar el algoritmo DeepAR.

Nombre de métrica Descripción Dirección de optimización
test:RMSE

La desviación cuadrática media entre la previsión y el destino real calculado en el conjunto de pruebas.

Minimizar

test:mean_wQuantileLoss

El promedio de pérdidas de cuantiles global calculado en el conjunto de pruebas. Para controlar qué cuantiles se utilizan, establezca el hiperparámetro test_quantiles.

Minimizar

train:final_loss

La pérdida de probabilidad de registro negativo de capacitación media en la última fecha de inicio de capacitación para el modelo.

Minimizar

Hiperparámetros ajustables para el algoritmo DeepAR

Ajuste un modelo DeepAR con los siguientes hiperparámetros. Los hiperparámetros que tienen el mayor impacto, en orden de mayor a menor impacto, en métricas objetivas de DeepAR: epochs, context_length, mini_batch_size, learning_rate y num_cells.

Nombre del parámetro Tipo de parámetro Intervalos recomendados
epochs

IntegerParameterRanges

MinValue: 1, 1000 MaxValue

context_length

IntegerParameterRanges

MinValue: 1, MaxValue 200

mini_batch_size

IntegerParameterRanges

MinValue: 32, MaxValue 1028

learning_rate

ContinuousParameterRange

MinValue: 1e-5, 1e-1 MaxValue

num_cells

IntegerParameterRanges

MinValue: 30, 200 MaxValue

num_layers

IntegerParameterRanges

MinValue: 1, MaxValue 8

dropout_rate

ContinuousParameterRange

MinValue: 0,00, MaxValue 0,2

embedding_dimension

IntegerParameterRanges

MinValue: 1, MaxValue 50