Machine Learning with Swift

Exporting the model for iOS

In our Jupyter notebook, execute the following code to export the model:

In []: 
import coremltools as coreml 
coreml_model = coreml.converters.sklearn.convert(tree_model, feature_names, 'label') 
coreml_model.author = "Author name goes here..." 
coreml_model.license = "License type goes here ..." 
coreml_model.short_description = "Decision tree classifier for extraterrestrials." 
coreml_model.input_description['data'] = "Extraterrestrials features" 
coreml_model.output_description['prob'] =  "Probability of belonging to class." 
coreml_model.save('DecisionTree.mlmodel') 

Scikit-learn converter documentation: http://pythonhosted.org/coremltools/generated/coremltools.converters.sklearn.convert.html#coremltools.converters.sklearn.convert ...