iOS11 新特性-Core ML (三) Core ML 实战

    上篇中我们已经生成了一个 Core ML 模型文件：HousePricer.mlmodel，本篇将介绍如何将文件引入到项目中并调用相应的接口。

1.      导入HousePricer.mlmodel到项目中
        1.1 文件导入方式同一般文件导入，采用拖拽或者菜单导入，选中导入的文件，我们可以查看模型的元数据及输入，输出信息(图一)
                                                             (图一)
   1.2 导入文件添加到编译资源
   上一步中导入的文件是不会自动添加到编译资源列表中(图二)，我们需要手动添加到编译资源中，否则无法调用接口
   
                                                            (图二)
   此时，我们再次选中 模型文件，在元数据信息面板，我们可以看到 在 Model class 部分 多了一个 可以点击的 向右箭头(图三)
   
                                                            (图三)
   点击 箭头，则进入 自动生成的 类文件 HousePricer.swift, 其中包含三个类：
   
   • HousePricerInput         输入参数
     
   • HousePricerOutput      输出参数
     
   • HousePricer                 用于预测接口的调用

2.    接口调用
   调用类中 生成 模型实例：
   

   ...
   let model = HousePricer()
   ...

   
   模型预测接口调用：
   模型中为我们生成了两个预测接口，代码如下：
   

       func prediction(input: HousePricerInput) throws -> HousePricerOutput {
           let outFeatures = try model.prediction(from: input)
           let result = HousePricerOutput(Price: outFeatures.featureValue(for: "Price")!.doubleValue)
           return result
       }
       func prediction(Square_Feet: Double) throws -> HousePricerOutput {
           let input_ = HousePricerInput(Square_Feet: Square_Feet)
           return try self.prediction(input: input_)
       }

   
   我们可以选择其中之一调用即可：
   

           let input = HousePricerInput(Square_Feet: size);
           guard let marsHabitatPricerOutput = try? model.prediction(input:input) else {
               fatalError("Unexpected runtime error.")
           }
           
           let price = marsHabitatPricerOutput.Price

   
   是不是很简单，完整代码示例，请移步Github