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