MLImageClassifier.ModelParameters
Parameters that affect the process of training an image classifier model.
Declaration
struct ModelParametersOverview
Use this structure to configure the model training session. With it you can:
Set a limit to the number of training iterations the session can use
Provide your own validation dataset. See MLImageClassifier.ModelParameters.ValidationData.
Enable specific image augmentations. See MLImageClassifier.ImageAugmentationOptions.
Designate a custom feature extractor. See MLImageClassifier.FeatureExtractorType.custom(_:).
Once you configure an MLImageClassifier.ModelParameters instance, use it to configure a training session with one of the applicable MLImageClassifier asynchronous type methods or synchronous initializers.
Topics
Creating parameters
init(validation:maxIterations:augmentation:algorithm:)init(featureExtractor:validation:maxIterations:augmentationOptions:)init(featureExtractor:validationData:maxIterations:augmentationOptions:)MLImageClassifier.ModelParameters.ClassifierTypeMLImageClassifier.ModelParameters.ModelAlgorithmType
Accessing the training parameters
Describing parameters
Supporting types
MLImageClassifier.FeatureExtractorTypeMLImageClassifier.ModelParameters.ValidationDataMLImageClassifier.ImageAugmentationOptions