无法使用CoreML模型进行预测

时间:2020-06-09 19:02:27

标签: swift coreml

我已经成功编译并保存了MLmodel,并且能够在快速代码中使用它。但是,当我尝试通过加载和编译使用同一模型时,会出现此错误。如果我打印模型,则会得到所有相关信息。但是,我无法做出预测。如果将错误放置在模型训练后的代码部分中,则抛出错误的“ let myprediction = try loadmodel.predictions(from:totestTable)”行实际上可以正常工作。

它必须是某种类型的声明/初始化。

func machinlearningTest( id : String){

    do {
        let compiledModelUrl: URL = try MLModel.compileModel(at: URL(string : "/Users/panos/Downloads/" + id + ".mlmodel")!)
        let loadedmodel  = try MLModel.init(contentsOf: compiledModelUrl)
        let testTable = try MLDataTable(contentsOf: URL(fileURLWithPath: "/Users/panos/Downloads/test.csv"))
        let testclassifierColumns = ["parameter1", "parameter2", "parameter3", "parameter4"]
        let totestTable = testTable[testclassifierColumns]
        let myprediction = try loadedmodel.predictions(from : totestTable)
        print(loadedmodel)

        }
    catch {}
 }

0 个答案:

没有答案