CNN回归分类：mnist数据集

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
import seaborn as sns
from keras.utils.np_utils import to_categorical
import tensorflow as tf
import tflearn.data_utils as du
from keras.models import Sequential
from keras.layers import Dense, Dropout, Flatten, Conv2D, MaxPool2D,Layer,Activation
from keras.optimizers import RMSprop
from keras.preprocessing.image import ImageDataGenerator
from sklearn.metrics import confusion_matrix

train = pd.read_csv('D:/sjk/mnist/mnist_train.csv', header = None)
train_data=train.iloc[:,1:785].values.astype('float32')
train_label=train.iloc[:,0].values.astype('int32')
test= pd.read_csv('D:/sjk/mnist/mnist_test.csv', header = None)
test_data=test.iloc[:,1:785].values.astype('float32')
test_label=test.iloc[:,0].values.astype('int32')

#归一化
train_data=train_data/255.0
te