Returned Y_train from load_train_data is all one

[ductri]

I think it may relate to python version, I'm using version 3.6 When I run x, y = load_train_data() x looks ok, but y is a matrix of 1 (with shape [73257, 10]).

Then I figure the problem comes from this line: Y_train = to_categorical(Y_train,10) I has changed it to: Y_train = to_categorical(Y_train[:, 0],10) and the problem is fixed.

Hope this helpful