Share via

ImageModelSettingsClassification Clase

  • Referencia

Configuración del modelo para las tareas de clasificación de imágenes de AutoML.

Herencia
azure.ai.ml.entities._job.automl.image.image_model_settings.ImageModelDistributionSettings
ImageModelSettingsClassification

Constructor

ImageModelSettingsClassification(*, advanced_settings: str | None = None, ams_gradient: bool | None = None, beta1: float | None = None, beta2: float | None = None, checkpoint_frequency: int | None = None, checkpoint_run_id: str | None = None, distributed: bool | None = None, early_stopping: bool | None = None, early_stopping_delay: int | None = None, early_stopping_patience: int | None = None, enable_onnx_normalization: bool | None = None, evaluation_frequency: int | None = None, gradient_accumulation_step: int | None = None, layers_to_freeze: int | None = None, learning_rate: float | None = None, learning_rate_scheduler: LearningRateScheduler | None = None, model_name: str | None = None, momentum: float | None = None, nesterov: bool | None = None, number_of_epochs: int | None = None, number_of_workers: int | None = None, optimizer: StochasticOptimizer | None = None, random_seed: int | None = None, step_lr_gamma: float | None = None, step_lr_step_size: int | None = None, training_batch_size: int | None = None, validation_batch_size: int | None = None, warmup_cosine_lr_cycles: float | None = None, warmup_cosine_lr_warmup_epochs: int | None = None, weight_decay: float | None = None, training_crop_size: int | None = None, validation_crop_size: int | None = None, validation_resize_size: int | None = None, weighted_loss: int | None = None, **kwargs)

Parámetros

advanced_settings
str
Requerido

Configuración de escenarios avanzados.

ams_gradient
bool
Requerido

Habilite AMSGrad cuando el optimizador sea "adam" o "adamw".

beta1
float
Requerido

Valor de "beta1" cuando el optimizador es "adam" o "adamw". Debe ser un valor float en el intervalo [0, 1].

beta2
float
Requerido

Valor de "beta2" cuando el optimizador es "adam" o "adamw". Debe ser un valor float en el intervalo [0, 1].

checkpoint_frequency
int
Requerido

Frecuencia para almacenar puntos de control del modelo. Debe ser un entero positivo.

checkpoint_run_id
str
Requerido

Identificador de una ejecución anterior que tiene un punto de control entrenado previamente para el entrenamiento incremental.

distributed
bool
Requerido

Si se va a usar el entrenamiento distribuido.

early_stopping
bool
Requerido

Habilite la lógica de detención temprana durante el entrenamiento.

early_stopping_delay
int
Requerido

Número mínimo de épocas o evaluaciones de validación que se deben esperar antes de que se realice un seguimiento de la mejora de la métrica principal para la detención temprana. Debe ser un entero positivo.

early_stopping_patience
int
Requerido

Número mínimo de épocas o evaluaciones de validación sin mejora de métrica principal antes de detener la ejecución. Debe ser un entero positivo.

enable_onnx_normalization
bool
Requerido

Habilite la normalización al exportar el modelo ONNX.

evaluation_frequency
int
Requerido

Frecuencia para evaluar el conjunto de datos de validación para obtener puntuaciones de métricas. Debe ser un entero positivo.

gradient_accumulation_step
int
Requerido

La acumulación de degradado significa ejecutar un número configurado de pasos "GradAccumulationStep" sin actualizar los pesos del modelo al acumular los degradados de esos pasos y, a continuación, usar los degradados acumulados para calcular las actualizaciones de peso. Debe ser un entero positivo.

layers_to_freeze
int
Requerido

Número de capas que se van a inmovilizar para el modelo. Debe ser un entero positivo. Por ejemplo, pasar 2 como valor para "seresnext" significa congelación de capa0 y capa1. Para obtener una lista completa de los modelos admitidos y detalles sobre la inmovilización de capas, consulte: https://docs.microsoft.com/en-us/azure/machine-learning/how-to-auto-train-image-models.

learning_rate
float
Requerido

Velocidad de aprendizaje inicial. Debe ser un valor float en el intervalo [0, 1].

learning_rate_scheduler
str o LearningRateScheduler
Requerido

Tipo de programador de velocidad de aprendizaje. Debe ser "warmup_cosine" o "paso". Entre los valores posibles se incluyen: "None", "WarmupCosine", "Step".

model_name
str
Requerido

Nombre del modelo que se va a usar para el entrenamiento. Para obtener más información sobre los modelos disponibles, visite la documentación oficial: https://docs.microsoft.com/en-us/azure/machine-learning/how-to-auto-train-image-models.

momentum
float
Requerido

Valor del impulso cuando el optimizador es "sgd". Debe ser un valor float en el intervalo [0, 1].

nesterov
bool
Requerido

Habilite nesterov cuando el optimizador sea "sgd".

number_of_epochs
int
Requerido

Número de épocas de entrenamiento. Debe ser un entero positivo.

number_of_workers
int
Requerido

Número de trabajos del cargador de datos. Debe ser un entero no negativo.

optimizer
str o StochasticOptimizer
Requerido

Tipo de optimizador. Entre los valores posibles se incluyen: "None", "Sgd", "Adam", "Adamw".

random_seed
int
Requerido

Inicialización aleatoria que se usará al usar el entrenamiento determinista.

step_lr_gamma
float
Requerido

Valor de gamma cuando el programador de velocidad de aprendizaje es "paso". Debe ser un valor float en el intervalo [0, 1].

step_lr_step_size
int
Requerido

Valor del tamaño del paso cuando el programador de velocidad de aprendizaje es "paso". Debe ser un entero positivo.

training_batch_size
int
Requerido

Tamaño del lote de entrenamiento. Debe ser un entero positivo.

validation_batch_size
int
Requerido

Tamaño del lote de validación. Debe ser un entero positivo.

warmup_cosine_lr_cycles
float
Requerido

Valor del ciclo de coseno cuando el programador de velocidad de aprendizaje es "warmup_cosine". Debe ser un valor float en el intervalo [0, 1].

warmup_cosine_lr_warmup_epochs
int
Requerido

Valor de las épocas de preparación cuando el programador de velocidad de aprendizaje es "warmup_cosine". Debe ser un entero positivo.

weight_decay
float
Requerido

Valor de la descomposición del peso cuando el optimizador es "sgd", "adam" o "adamw". Debe ser un valor flotante en el intervalo[0, 1].

training_crop_size
int
Requerido

Tamaño de recorte de imagen que se introduce en la red neuronal para el conjunto de datos de entrenamiento. Debe ser un entero positivo.

validation_crop_size
int
Requerido

Tamaño de recorte de imagen que se introduce en la red neuronal para el conjunto de datos de validación. Debe ser un entero positivo.

validation_resize_size
int
Requerido

Tamaño de imagen al que se va a cambiar el tamaño antes de recortar para el conjunto de datos de validación. Debe ser un entero positivo.

weighted_loss
int
Requerido

Pérdida ponderada. Los valores aceptados son 0 sin pérdida ponderada. 1 para pérdida ponderada con sqrt. (class_weights). 2 para pérdida ponderada con class_weights. Debe ser 0, 1 o 2.