Get started with Create ML to train a machine learning model in iOS
Learn how to use Create ML to create machinge learning models.
03 Apr 2023 · 4 min read
As mentioned in this developer guide on machine learning for iOS with Core ML, Apple's Create ML app allows us to train a model.
Create ML is part of Xcode's tool chain and allows us to train machinge learning models without requiring extensive knowledge of machine learning, for example allowing us to drag and drop datasets, choose an algorithm, and configure the model parameters through a graphical interface.
Let's look at how to do that.
After opening Create ML, the application presents different templates to choose from. They can be used to train models for tasks such as image classification, object detection, text and sound classification, classification and regression of tabular data and more.
As an example, we'll work with the image classification template.
Let's say, we want to train a model which recognizes different weather conditions like rain, sunshine, sunset and clouds in photos.
To be able to train the model to do that, we need a labeled dataset of photos. The data set might look as follows:
As we can see above, the images are categorized in folders. The folder names represent the image labels.
After adding the data set, Create ML creates a class for each folder:
This is all we need to do to be able to start the training process by clicking the train button.
After the model training is completed, we can test the model with new data in the preview tab.
When satisfied with the model, we can save it as a .mlmodel file from the output tab and integrate it into our iOS application. To learn how to do that, check out this article on machine learning for iOS with Core ML.
Newsletter
Like to support my work?
Say hi
Related tags
Articles with related topics
Latest articles and tips