Create ML学习笔记

Create ML 目前版本可以创建两类模型:图片分类模型、文本分类模型。

运行环境:
macOS 10.14 Mojave
Xcode 10.x

创建图片分类器

数据准备:
训练数据集(training dataset) 、测试数据集(testing dataset)
数据集中子文件夹即分类,子文件夹的名字即类别名称,不包含标签文件。

创建方式
在Xcode 10中,创建 macOS 版的 playground,输入下面代码:

import CreateMLUI

let builder = MLImageClassifierBuilder()
builder.showInLiveView()

点击运行按钮,在assistant editor中会出现交互式视图。

Create ML学习笔记_第1张图片

首先可以配置参数

Create ML学习笔记_第2张图片

Max iterations:设置最大迭代数
Augmentation:通过对图片进行一些变换来增加不同状态的数据量,要根据使用需求设置。

把训练数据集拖拽到虚线框中,训练自动开始,我们可以在debug区域观察 Training 及 Validation 准确率指标。

训练完成后拖拽测试数据集开始测试,查看测试结果,如果满意可以把模型保存下来。

Create ML学习笔记_第3张图片

最终生成的.mlmodel文件非常小只有20k左右。

你可能感兴趣的:(Create ML学习笔记)