coreML故障验证输入

Question

let model = test2()
        var data = [1.0, 2.0,2.0,2.0,2.0,2.0,2.0]

        guard let mlMultiArray = try? MLMultiArray(shape:[1,7], dataType:MLMultiArrayDataType.double) else {
            fatalError("Unexpected runtime error. MLMultiArray")
        }

        for (index, element) in data.enumerated() {
            mlMultiArray[index] = NSNumber(floatLiteral: element)
        }





        guard let markupOut = try? model.prediction(input1: mlMultiArray) else {
            fatalError("Unexpected runtime error.")
        }

我使用的是斯威夫特4和核心ML。

编译但在运行时失败：

coreml输入input1是2级数组，但该模型只支持单个向量输入(1级)或一批向量序列(3级)。coreml失败验证输入。

input1是MultiArray的类型(双7)

我是否通过将双变量的MultiArray转换为单个向量输入/批向量来修正“失败验证输入”错误？如果是这样的话，我如何将双倍的MultiArray转换为单个向量输入或向量批次？

Answer 1

当您编写shape:[1,7]时，MLMultiArray为2级，核心ML表示该模型不支持第2级输入。因此，shape:[7] (使其排名为1)或shape:[1,1,7] (使其排名第3)都是如此。