Scikit-learn

You can convert a scikit-learn pipeline, classifier, or regressor to the Core ML format:

from sklearn.linear_model import LinearRegression import pandas as pd # Load data data = pd.read_csv('houses.csv') # Train a model model = LinearRegression() model.fit(data[["bedroom", "bath", "size"]], data["price"]) # Convert and save the scikit-learn model import coremltools as ct coreml_model = ct.converters.sklearn.convert( model, ["bedroom", "bath", "size"], "price") coreml_model.save('HousePricer.mlmodel')

For more information, see the API reference.


Did this page help you?