美文网首页
Create ML学习笔记

Create ML学习笔记

作者: 梁间 | 来源:发表于2018-09-26 16:23 被阅读0次

    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左右。

    相关文章

      网友评论

          本文标题:Create ML学习笔记

          本文链接:https://www.haomeiwen.com/subject/gitjoftx.html