How to change the dimensions of the original dataset to get the same forecast value? LTSM ONNX

Question

slevin Lee 2022-11-18

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/1854768-how-to-change-the-dimensions-of-the-original-dataset-to-get-the-same-forecast-value-ltsm-onnx

回答： Sivylla Paraskevopoulou 2022-11-18

采纳的回答： Sivylla Paraskevopoulou

I trained an LSTM model in matlab

The original data set is 56 * 3853

feature_dimension is 56

miniBatchSize = 50;

[net,YPred] = predictAndUpdateState(net,xtrain,'ExecutionEnvironment','cpu');

normal operation

then...

filename = "lstm.onnx";

exportONNXNetwork(net,filename)

save data.mat xtrain

in python...

import onnxruntime

onnx_model = onnxruntime.InferenceSession('lstm.onnx')

from scipy import io

mydata = io.loadmat('data.mat')

data=np.float32(mydata['xtrain'])

onnx_input = {onnx_model.get_inputs()[0].name: data}

outputs = onnx_model.run(None, onnx_input)

InvalidArgument: [ONNXRuntimeError] : 2 : INVALID_ARGUMENT : Invalid rank for input: input Got: 2 Expected: 3 Please fix either the inputs or the model.

How to change the dimensions of the original dataset to get the same forecast value? LTSM ONNX

thanks a lot !!!

0 个评论
显示 -2更早的评论隐藏 -2更早的评论

请先登录，再进行评论。

请先登录，再回答此问题。

Answer 1

Sivylla Paraskevopoulou 2022-11-18

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/1854768-how-to-change-the-dimensions-of-the-original-dataset-to-get-the-same-forecast-value-ltsm-onnx#answer_1104268

You should permute from the MATLAB ordering (CN) to the ONNX ordering (NC), where C is the number of features and N is the number of sequence observations. Permute the input data before saving them to data.mat.

xtrain = permute(xtrain,[2,1]);

For more information about dimension ordering for different deep learning platforms, see Input Dimension Ordering.