使用 CreateML 应用创建 CoreML 模型的指南 — 图像分类

CoreML 框架使得将机器学习模型集成到 Apple 平台应用中变得更加容易。

虽然有多种方法可以创建 CoreML 模型，但使用 Xcode 中的 CreateML 开发者工具无疑是最简单的方法之一。

我们使用一个著名的猫狗数据集，来展示如何使用 CreateML 创建图像分类模型。

数据集下载地址：Microsoft 下载链接

1. 启动 CreateML 应用

首先，打开 Xcode。然后在菜单栏中点击 Xcode，选择“打开开发者工具”，然后点击“CreateML”。此时会出现一个查找器窗口，选择新文档以启动一个新的 CreateML 项目。

在模板选择页面中，选择图像分类模型类型，然后点击“下一步”。

输入模型的名称和作者，然后选择一个合适的文件夹来保存 CreateML 项目。

在训练机器学习模型时，设置页面可能看起来有些复杂，但实际上非常简单。

在数据部分，有三个部分：训练数据、验证数据和测试数据。训练数据用于训练模型，验证数据用于评估模型在训练过程中的表现，测试数据用于最终评估模型的准确性。

在参数部分，主要设置最大迭代次数，即训练过程中数据通过模型的最大次数。

在增强部分，可以在数据传递给模型之前进行一些编辑，以防止模型过拟合。

将猫狗数据集按照要求的布局组织好，拖到训练数据部分。由于数据量大，加载时间可能较长。

无需数据扩充，点击窗口左上角的播放按钮开始训练。训练时间因 Mac 的性能而异。

训练完成后，评估页面显示每个类别的准确率和召回率。这些指标用于评估模型的性能。

预览页面允许您使用自己的图像测试模型，查看预测结果。

训练完成后，可以导出模型以便在应用中使用。点击输出页面中的“获取”按钮，或者直接发送到 Xcode 项目。