How to initialize a MLMultiArray in CoreML

前端 未结 2 1673
被撕碎了的回忆
被撕碎了的回忆 2021-01-18 17:32

I\'ve got an array of 40 arrays with 12 double features, so the type is [[double]]. Currently I\'m sending this data to Google Cloud ML API to get a related prediction.

2条回答
  •  迷失自我
    2021-01-18 18:16

    I achieved it by reading this blog :)

    let data = _currentScaledMotionArrays.reduce([], +) //result is of type [Double] with 480 elements
    guard let mlMultiArray = try? MLMultiArray(shape:[40,12], dataType:MLMultiArrayDataType.double) else {
        fatalError("Unexpected runtime error. MLMultiArray")
    }
    for (index, element) in data.enumerated() {
        mlMultiArray[index] = NSNumber(floatLiteral: element)
    }
    let input = PredictionModelInput(accelerations: mlMultiArray)
    guard let predictionOutput = try? _predictionModel.prediction(input: input) else {
            fatalError("Unexpected runtime error. model.prediction")
    }
    

提交回复
热议问题