I hope someone may help me with my error message. I am running the onnxruntime sess = rt.InferenceSession()
and then res = sess run(None, {a dictionary of things I want})
. When I print the result, I get an array that looks like this,
[array([[0.86897475, 0.13102525],
[0.68731904, 0.31268096],
[0.68731904, 0.31268096],
...,
[0.68731904, 0.31268096],
[0.86897475, 0.13102525],
[0.7062882 , 0.29371178]], dtype=float32)]
I want to convert this into a dataframe, so I use the following,
prob = pd.DataFrame(res[1][:,1], columns=['prob_1'])
However, I get an error message: IndexError: list index out of range.
I want to build a dataframe of the second column, and call it prob_1.
I hope the above made sense and would appreciate any help.
Thank you
1