text_html_output_1.png
import pandas as pd

train_data = pd.read_csv('/kaggle/input/digit-recognizer/train.csv')

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

train_data.shape

train_data = train_data.values.reshape(-1, 28, 28, 1)

train_data.shape