from keras.utils import to_categorical # 导入keras.utils工具库的类别转换工具
X_train = X_train_image.reshape(60000, 28, 28, 1) # 给标签增加一个维度
X_test = X_test_image.reshape(10000, 28, 28, 1) # 给标签增加一个维度
y_train = to_categorical(y_train_lable, 10) # 特征转换为one-hot编码
y_test = to_categorical(y_test_lable, 10) # 特征转换为one-hot编码
print ("训练集张量形状：", X_train.shape) # 训练集张量的形状
print ("第一个数据标签：", y_train[0]) # 显示标签集的第一个数据