Ajuste de un modelo Text Classification - TensorFlow
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 contabilizadas por el algoritmo Text Classification - TensorFlow
Consulte la siguiente tabla para ver qué métricas calcula el algoritmo Text Classification - TensorFlow.
Nombre de métrica | Descripción | Dirección de optimización | Patrón de expresiones regulares |
---|---|---|---|
validation:accuracy |
La ratio del número de predicciones correctas con respecto al número total de predicciones realizadas. |
Maximizar |
|
Hiperparámetros ajustables de Text Classification - TensorFlow
Puede ajustar un modelo de clasificación de texto con los siguientes hiperparámetros. Los hiperparámetros con un mayor impacto en las métricas de objetivos de clasificación de texto son batch_size
, learning_rate
y optimizer
. Ajuste los hiperparámetros relacionados con el optimizador, como momentum
, regularizers_l2
, beta_1
, beta_2
y eps
, en función del optimizer
seleccionado. Por ejemplo, utilice beta_1
y beta_2
solo cuando el optimizer
sea adamw
o adam
.
Para obtener más información sobre los hiperparámetros que se utilizan en cada optimizer
, consulte Hiperparámetros de Text Classification - TensorFlow.
Nombre del parámetro | Tipo de parámetro | Intervalos recomendados |
---|---|---|
batch_size |
IntegerParameterRanges |
MinValue: 4; MaxValue: 128 |
beta_1 |
ContinuousParameterRanges |
MinValue: 1e-6, MaxValue: 0,999 |
beta_2 |
ContinuousParameterRanges |
MinValue: 1e-6, MaxValue: 0,999 |
eps |
ContinuousParameterRanges |
MinValue: 1e-8, MaxValue: 1,0 |
learning_rate |
ContinuousParameterRanges |
MinValue: 1e-6, MaxValue: 0,5 |
momentum |
ContinuousParameterRanges |
MinValue: 0,0, MaxValue: 0,999 |
optimizer |
CategoricalParameterRanges |
['adamw', 'adam', 'sgd', 'rmsprop', 'nesterov', 'adagrad', 'adadelta'] |
regularizers_l2 |
ContinuousParameterRanges |
MinValue: 0,0, MaxValue: 0,999 |
train_only_on_top_layer |
CategoricalParameterRanges |
['True', 'False'] |