In [58]:
# Logistic Regression
from sklearn import datasets
from sklearn import metrics
from sklearn.linear_model import LogisticRegression
In [59]:
# load the iris datasets
dataset = datasets.load_iris()
In [60]:
# fit a logistic regression model to the data
model = LogisticRegression()
model.fit(dataset.data, dataset.target)
Out[60]:
In [61]:
coreml_model = coremltools.converters.sklearn.convert(model,["a", "b", "c", "d"],"y")
In [62]:
coreml_model.save('Iris.mlmodel')