application_vnd.jupyter.stderr_output_1.png
import numpy as np # linear algebra

import pandas as pd # data processing, CSV file I/O (e.g. pd.read_csv)

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

x_train = train_data.drop(labels='label', axis=1)

x_train = x_train / 255
test_data = test_data / 255

X_train = x_train.values.reshape(x_train.shape[0], 28, 28, 1)
X_test = test_data.values.reshape(test_data.shape[0], 28, 28, 1)

print(X_train.shape)
print(X_test.shape)