Instantly share code, notes, and snippets.

Embed
What would you like to do?
// Parámetros
let parameter = MLImageClassifier.ModelParameters(featureExtractor: MLImageClassifier.FeatureExtractorType.scenePrint(revision: 1),
validationData: validationDataSource,
maxIterations: 10,
augmentationOptions: [ .flip, .rotation ])
// Creamos el clasificador
let classifier = try? MLImageClassifier(trainingData: trainingDataSource, parameters: parameter)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment