import numpy as np
from keras.datasets import mnist
from keras.utils import np_utils
from keras.models import Sequential
from keras.layers import Dense,Dropout
from keras.optimizers import SGD
#load data(从网上下载数据集!)
(x_train,y_train),(x_test,y_test)=mnist.load_data()
#x_shape: (60000, 28, 28)
print('x_shape:',x_train.shape)
#y_shape: (60000,)
print('y_shape:',y_train.shape)
#x_shape: (60000, 28, 28)->x_shape: (60000, 784)
x_train=x_train.reshape(x_train.shape[0],784)/255.0#shape0就是60000,-1自动计算28*28
x_test=x_test.reshape(x_test.shape[0],784)/255.0
#换one hot格式:把输出训练成10个类
y_train=np_utils.to_categorical(y_train,num_classes=10)
y_test=np_utils.to_categorical(y_test,num_classes=10)
#创建模型:输入784个神经元,输出200个神经元
model=Sequential([
Dense(units=200,input_dim=784,bias_initializer='one',activation='tanh'),
keras速度复习-dropout与正则化
最新推荐文章于 2022-03-30 22:25:38 发布