Skip to content

Instantly share code, notes, and snippets.

@sahil280114
Created August 16, 2018 09:26
Show Gist options
  • Save sahil280114/6dbfc5cce45e195ab0d5d218d28a71d2 to your computer and use it in GitHub Desktop.
Save sahil280114/6dbfc5cce45e195ab0d5d218d28a71d2 to your computer and use it in GitHub Desktop.
coreml_model = coremltools.converters.sklearn.convert(model)
coreml_model.author = 'Your Name'
coreml_model.license = 'MIT'
coreml_model.short_description = 'Sentiment polarity LinearSVC.'
coreml_model.input_description['input'] = 'Features extracted from the text.'
coreml_model.output_description['classLabel'] = 'The most likely polarity (positive or negative), for the given input.'
coreml_model.output_description['classProbability'] = 'The probabilities for each class label, for the given input.'
coreml_model.save('SentimentPolarity.mlmodel')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment