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中会出现交互式视图。
首先可以配置参数
Max iterations:设置最大迭代数
Augmentation:通过对图片进行一些变换来增加不同状态的数据量,要根据使用需求设置。
把训练数据集拖拽到虚线框中,训练自动开始,我们可以在debug区域观察 Training 及 Validation 准确率指标。
训练完成后拖拽测试数据集开始测试,查看测试结果,如果满意可以把模型保存下来。
最终生成的.mlmodel文件非常小只有20k左右。
网友评论