在swift中将.npy转换为MLMultiArray以进行CoreML预测

import PythonKit

let np = Python.import("numpy")
let array = np.load("path/to/file.npy")

let arrayData = Array<Double>(array)

guard let multiArray = try? MLMultiArray(shape: [NSNumber(value: arrayData.count)], dataType: .double) else {
    fatalError("Failed to create MLMultiArray.")
}

for (index, element) in arrayData.enumerated() {
    multiArray[index] = NSNumber(value: element)
}

import CoreML
import Accelerate
import PythonKit

// Step 1: Load .npy file using PythonKit
let np = Python.import("numpy")
let array = np.load("path/to/file.npy")

// Step 2: Convert NumPy array to Swift Array
let arrayData = Array<Double>(array)

// Step 3: Create MLMultiArray and copy data
guard let multiArray = try? MLMultiArray(shape: [NSNumber(value: arrayData.count)], dataType: .double) else {
    fatalError("Failed to create MLMultiArray.")
}

for (index, element) in arrayData.enumerated() {
    multiArray[index] = NSNumber(value: element)
}

// Step 4: Use MLMultiArray for CoreML prediction
// ...