Is there a way to make NDArrayIter spit out integer labels? It always converts labels to float, and in the example below I can’t convert back as some significant digits get lost. Thanks!

from mxnet import nd

import numpy as np

X_train = np.array([1,2,3,4,5])

Y_train = np.array([1,2,3,4,5111111122])

train_data = NDArrayIter(X_train, Y_train, 2, shuffle=False)

train_data.label[0]

Output:

(‘softmax_label’,

[ 1.00000000e+00 2.00000000e+00 3.00000000e+00 4.00000000e+00

5.11111117e+09]

<NDArray 5 @cpu(0)>)