MLMultiArray是否是模型输出矩阵的扁平数组?

初次接触CoreML,正在尝试理解一些基本概念。

我正在处理一个输出为以下格式的模型:

float32 [1,896,16]

在使用模型并将输出获取为MLMultiArray时,我得到以下结果:

let output = prediction.regressors // MLMultiArrayprint(output.debugDescription) // Float32 1 x 896 x 16 arrayprint(output.count) // 14336, 这是896x16

我可以简单地使用output[0]..output[1]..来访问每个元素

对于存储在MLMultiArray中的任何数据类型,这都是正确的吗?这是Swift提供给我们的“便利”吗?

如果是扁平数组的情况,它的顺序会与矩阵的顺序相同吗?


回答:

MLMultiArray是否是模型输出矩阵的扁平数组?不是,它是给定维度的多维数组

您可以像下面这样将其转换为数组,并且它应该与矩阵的顺序相同。

let length = output.count    let doublePtr =  output.dataPointer.bindMemory(to: Double.self, capacity: length)    let doubleBuffer = UnsafeBufferPointer(start: doublePtr, count: length)    let outputArray = Array(doubleBuffer)

关于数据类型,苹果的文档仅显示三种类型:这里

case int32

代表多维数组的整数类型,通常用于文本编码。

case float32

代表多维数组中的浮点类型。

case double

代表多维数组的双精度类型。

Related Posts

使用LSTM在Python中预测未来值

这段代码可以预测指定股票的当前日期之前的值,但不能预测…

如何在gensim的word2vec模型中查找双词组的相似性

我有一个word2vec模型,假设我使用的是googl…

dask_xgboost.predict 可以工作但无法显示 – 数据必须是一维的

我试图使用 XGBoost 创建模型。 看起来我成功地…

ML Tuning – Cross Validation in Spark

我在https://spark.apache.org/…

如何在React JS中使用fetch从REST API获取预测

我正在开发一个应用程序,其中Flask REST AP…

如何分析ML.NET中多类分类预测得分数组?

我在ML.NET中创建了一个多类分类项目。该项目可以对…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注