Partager via


ImageClassificationTrainer.Options Classe

Définition

Classe Options pour ImageClassificationTrainer.

public sealed class ImageClassificationTrainer.Options : Microsoft.ML.Trainers.TrainerInputBaseWithLabel
type ImageClassificationTrainer.Options = class
    inherit TrainerInputBaseWithLabel
Public NotInheritable Class ImageClassificationTrainer.Options
Inherits TrainerInputBaseWithLabel
Héritage
ImageClassificationTrainer.Options

Constructeurs

ImageClassificationTrainer.Options()

Classe Options pour ImageClassificationTrainer.

Champs

Arch

Spécifie l’architecture de modèle à utiliser dans le cas d’une formation à la classification d’images à l’aide de l’apprentissage par transfert. L’architecture par défaut est Resnet_v2_50.

BatchSize

Nombre d’échantillons à utiliser pour l’entraînement par mini-lots. La valeur par défaut de BatchSize est 10.

EarlyStoppingCriteria

Paramètres de technique d’arrêt précoce à utiliser pour arrêter l’entraînement lorsque la métrique d’entraînement cesse de s’améliorer. Par défaut, l’arrêt anticipé est activé et la métrique de surveillance est Précision.

Epoch

Nombre d’itérations d’entraînement. La valeur par défaut pour Epoch est 200.

FeatureColumnName

Colonne à utiliser pour les fonctionnalités.

(Hérité de TrainerInputBase)
FinalModelPrefix

Fichiers de modèle final et de point de contrôle/préfixe de dossier pour le stockage des fichiers de graphe. Le préfixe par défaut est « custom_retrained_model_based_on_ ».

LabelColumnName

Colonne à utiliser pour les étiquettes.

(Hérité de TrainerInputBaseWithLabel)
LearningRate

Taux d’apprentissage à utiliser pendant l’optimisation. La valeur par défaut du taux d’apprentissage est 0,01.

LearningRateScheduler

Classe qui effectue la planification du taux d’apprentissage. Le planificateur de taux d’apprentissage par défaut est la décroissance exponentielle du taux d’apprentissage.

MetricsCallback

Rappel pour signaler des statistiques sur la précision/l’entropie croisée pendant la phase d’entraînement. Le rappel de métriques est défini sur null par défaut.

PredictedLabelColumnName

Nom du tenseur qui contiendra l’étiquette prédite à partir des scores de sortie de la dernière couche lorsque l’apprentissage de transfert est terminé. Le nom tensoriel par défaut est « PredictedLabel ».

ReuseTrainSetBottleneckCachedValues

Indique de ne pas calculer à nouveau les valeurs de jeu de train de goulot d’étranglement mis en cache si elles sont déjà disponibles dans le dossier bin. Ce paramètre a la valeur false par défaut.

ReuseValidationSetBottleneckCachedValues

Indique de ne pas calculer à nouveau les valeurs du jeu de validation de goulot d’étranglement mis en cache si elles sont déjà disponibles dans le dossier bin. Ce paramètre a la valeur false par défaut.

ScoreColumnName

Nom du tenseur qui contiendra les scores de sortie de la dernière couche lorsque l’apprentissage de transfert est terminé. Le nom tensoriel par défaut est « Score ».

TestOnTrainSet

Indique d’évaluer le modèle sur le jeu d’apprentissage après chaque époque. Test sur la ramset est défini sur true par défaut.

TrainSetBottleneckCachedValuesFileName

Indique le nom de fichier dans l’espace de travail pour stocker les valeurs de goulot d’étranglement du train pour la mise en cache, le nom de fichier par défaut est « trainSetBottleneckFile.csv ».

ValidationSet

Jeu de validation.

ValidationSetBottleneckCachedValuesFileName

Indique le nom de fichier dans l’espace de travail pour stocker les valeurs de goulot d’étranglement du jeu de validation pour la mise en cache, le nom de fichier par défaut est « validationSetBottleneckFile.csv ».

ValidationSetFraction

Lorsque le jeu de validation n’est pas passé, une fraction du jeu de train est utilisée comme validation. Pour désactiver ce comportement, définissez sur ValidationSetFraction null. Accepte la valeur comprise entre 0 et 1,0, la valeur par défaut est 0,1 ou 10 % de la rame.

WorkspacePath

Indique le chemin d’accès où les fichiers de cache de goulot d’étranglement d’image et le modèle entraîné sont enregistrés. La valeur par défaut est un nouveau répertoire temporaire.

S’applique à