Keras学习笔记二

阿莫Keras学习笔记-Classifier分类（程序里面部分错误已更改，目前Keras采用API 2接口函数）

程序：

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, Activation

from keras.optimizers import RMSprop

(X_train, y_train), (X_test, y_test) = mnist.load_data()


#归一化
X_train = X_train.reshape(X_train.shape[0], -1) / 255
X_test = X_test.reshape(X_test.shape[0], -1)/255
y_train = np_utils.to_categorical(y_train, num_classes = 10)
y_test = np_utils.to_categorical(y_test, num_classes = 10)


model = Sequential([
    Dense(32, input_dim=784),
    Activation('relu'),
    Dense(10,),
    Activation('softmax'),       
])


#优化器
rmsprop = RMSprop(lr = 0.01, rho=0.9, epsilon=1e-08, decay=0.0)




model.compile(
   optimizer=rmsprop,
   loss='categorical_crossentropy',
   metrics=['accuracy'] 
)


print('Training --------------')
#训练模型
model.fit(X_train, y_train, batch_size=32, epochs=2)


print('\nTesting --------------')
#测试模型
loss,accuracy = model.evaluate(X_test, y_test)


print('test loss: ', loss)

print('test accuracy: ', accuracy)