面试题：Swift中CoreML集成的基础步骤

Question

请简述在Swift项目中集成CoreML模型的基本步骤，包括模型导入、实例化以及使用模型进行预测的关键代码片段。

Answer 1

1. 模型导入：
   • 将训练好的CoreML模型（.mlmodel文件）拖入Xcode项目中。在文件属性检查器中，确保“Target Membership”选择了对应的Swift项目目标。
2. 实例化模型：
   • 导入CoreML框架：

   import CoreML
   

   • 假设模型名称为MyModel，实例化模型：

   if let model = try? MyModel(configuration:.init()) {
       // 模型实例化成功
   } else {
       // 处理模型实例化失败情况
   }
   

3. 使用模型进行预测：
   • 假设模型有输入参数input，类型为String，并且返回一个预测结果output，类型为Double。首先创建输入对象：

   let input = MyModelInput(input: "your input string")
   

   • 然后进行预测：

   if let prediction = try? model.prediction(input: input) {
       let result = prediction.output
       print("预测结果: \(result)")
   } else {
       // 处理预测失败情况
   }