动态加载mlmodel

Question

我正在体验CoreML的项目能力。以下是我所做的：

1. 使用scikit创建.pkl文件-学习Python
2. 使用coremltools包将其转换为.mlmodel文件
3. 下载到我的iOS应用程序
4. 在运行时编译它： 让classifierName = "classifier1“让fileName =NSString(格式：”%@.mlmodel“，classifierName)让documentsUrl:URL = FileManager.default.urls(for：.documentDirectory，in：.userDomainMask).first作为URL！让destinationFileUrl = documentsUrl.appendingPathComponent(fileName作为字符串)让compiledModelUrl = try？MLModel.compileModel(at: destinationFileUrl)让模型=尝试？MLModel(contentsOf: compiledModelUrl!)

现在，我想用我的模型来做预测。我尝试在一个示例应用程序中直接嵌入.mlmodel文件，这允许XCode在构建时创建一个包装类来实例化输入：

let multiArr = try? MLMultiArray.init(shape: [1], dataType: .double)
let input = classifier1Input(input: multiArr!)
let output = try? model.prediction(input: input)

但是，由于我在运行时从服务器下载文件，所以无法访问此类包装器类。

let predict = model?.prediction(from: <MLFeatureProvider>)

有什么想法吗？

Answer 1

最简单的解决方案:将Xcode生成的包装类复制到Swift文件中，并将其添加到项目中。(请注意，这个包装类还展示了如何制作MLFeatureProvider等。)