Error when running the example cnn_mnist.py and mlp_mnist.py

[Zehui127]

Thanks for your great work I found some problem when running the example, I can't find the reason when running the above two example, the following error pop up:

Traceback (most recent call last): File "cnn_mnist.py", line 72, in run() File "cnn_mnist.py", line 62, in run nn.fit(X_train, y_train, learning_rate=0.05, max_iter=3, batch_size=32) File "F:\summerVacation\lib\site-packages\nnet\neuralnetwork.py", line 29, in fit Y_one_hot = one_hot(Y) File "F:\summerVacation\lib\site-packages\nnet\helpers.py", line 9, in one_hot one_hot_labels[labels == c, c] = 1 IndexError: only integers, slices (:), ellipsis (...), numpy.newaxis (None) and integer or boolean arrays are valid indices

I'm not sure why this happen.

[iamyup]

Hi,

I also have same problem. no one find the solution ?

[snowkylin]

@Zehui127 @iamyup In cnn_mnist.py and mlp_mnist.py, The type of y_train and y_test need to be converted to int, since the index of array cannot be float. i.e., modify the code y_train = mnist.target[:split] y_test = mnist.target[split:] to y_train = mnist.target[:split].astype("int") y_test = mnist.target[split:].astype("int")

[abhinav0491]

@snowkylin I tried the solution you provided, but Im still getting the same error.

Traceback (most recent call last): File "cnn_mnist.py", line 72, in run() File "cnn_mnist.py", line 62, in run nn.fit(X_train, y_train, learning_rate=0.05, max_iter=3, batch_size=32) File "/home/abhinav/Testing/nnet-master/nnet/neuralnetwork.py", line 29, in fit Y_one_hot = one_hot(Y) File "/home/abhinav/Testing/nnet-master/nnet/helpers.py", line 9, in one_hot one_hot_labels[labels == c, c] = 1 IndexError: only integers, slices (:), ellipsis (...), numpy.newaxis (None) and integer or boolean arrays are valid indices