text_plain_output_1.png
from keras.utils.np_utils import to_categorical

import pandas as pd

train = pd.read_csv('../input/train.csv')
test = pd.read_csv('../input/test.csv')

train_images = train.ix[:, 1:].values.astype('float32')
train_labels = train.ix[:, 0].values.astype('int32')

train_labels.shape

from keras.utils.np_utils import to_categorical
train_labels = to_categorical(train_labels)

train_labels.shape

train_labels[0:10]