init(featureExtractor:validation:maxIterations:augmentationOptions:)
Creates a new set of training parameters for an image classifier with a validation dataset.
Declaration
init(featureExtractor: MLImageClassifier.FeatureExtractorType = .scenePrint(revision: 1), validation: MLImageClassifier.ModelParameters.ValidationData = .split(strategy: .automatic), maxIterations: Int = 25, augmentationOptions: MLImageClassifier.ImageAugmentationOptions = [])Parameters
- featureExtractor:
The feature extractor you want Create ML to use during the training session.
- validation:
A validation dataset.
- maxIterations:
The maximum number of training iterations to use during training. The default is 25.
- augmentationOptions:
The image augmentation options to use to increase the training data variety.