Skip to content

Instantly share code, notes, and snippets.

@fitomad
Created October 24, 2018 14:26
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save fitomad/7e896f506b39a83da443e5ee06c50a02 to your computer and use it in GitHub Desktop.
Save fitomad/7e896f506b39a83da443e5ee06c50a02 to your computer and use it in GitHub Desktop.
// 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