text_plain_output_1.png
import pandas as pd

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

x_train = train.iloc[:, 1:].values / 255.0
y_train = train.iloc[:, 0].values
x_test = test.values / 255
print((x_train.shape, y_train.shape), x_test.shape)