makeTrainingSession(trainingData:parameters:sessionParameters:)
Creates or restores a training session.
Declaration
static func makeTrainingSession(trainingData: MLImageClassifier.DataSource, parameters: MLImageClassifier.ModelParameters = ModelParameters(
validation: .split(strategy: .automatic),
augmentation: [],
algorithm: .transferLearning(
featureExtractor: .scenePrint(revision: 1),
classifier: .logisticRegressor
)
), sessionParameters: MLTrainingSessionParameters = _defaultSessionParameters) throws -> MLTrainingSession<MLImageClassifier>Parameters
- trainingData:
Data source for training.
- parameters:
Model training parameters. See Modelparameters Swift.struct for the defaults.
- sessionParameters:
Training session parameters. See Mltrainingsessionparameters for the defaults.
Return Value
A MLTrainingSession that can be used to start or resume training.