keras mnist 分类

最新推荐文章于 2022-07-05 14:43:50 发布

阿豪boy

最新推荐文章于 2022-07-05 14:43:50 发布

阅读量54

点赞数

文章标签：机器学习深度学习 tensorflow python 神经网络

本文链接：https://blog.csdn.net/qq_35516360/article/details/122066094

版权

原文链接: keras mnist 分类

上一篇: keras 线性回归

下一篇: keras cnn 网咯 mnist 分类

可以使用list构建模型

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

构建模型，设置loss和需要的参数

rmsprop = RMSprop(lr=0.001, rho=0.9, epsilon=1e-08, decay=0.0)

# We add metrics to get more results you want to see
model.compile(optimizer=rmsprop,
              loss='categorical_crossentropy',
              metrics=['accuracy'])

参考代码

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

# download the mnist to the path '~/.keras/datasets/' if it is the first time to be called
# X shape (60,000 28x28), y shape (10,000, )
(X_train, y_train), (X_test, y_test) = mnist.load_data()
print(X_train.shape, y_train.shape)  # (60000, 28, 28) (60000,)
# data pre-processing
X_train = X_train.reshape(X_train.shape[0], -1) / 255.  # normalize
X_test = X_test.reshape(X_test.shape[0], -1) / 255.  # normalize
y_train = np_utils.to_categorical(y_train, num_classes=10)
y_test = np_utils.to_categorical(y_test, num_classes=10)
print(X_train.shape, y_train.shape)  # (60000, 784) (60000, 10)

# Another way to build your neural net
model = Sequential([
    Dense(32, input_dim=784),
    Activation('relu'),
    Dense(10),
    Activation('softmax'),
])

# Another way to define your optimizer
rmsprop = RMSprop(lr=0.001, rho=0.9, epsilon=1e-08, decay=0.0)

# We add metrics to get more results you want to see
model.compile(optimizer=rmsprop,
              loss='categorical_crossentropy',
              metrics=['accuracy'])

print('Training ------------')
# Another way to train the model
model.fit(X_train, y_train, epochs=2, batch_size=32)

print('\nTesting ------------')
# Evaluate the model with the metrics we defined earlier
loss, accuracy = model.evaluate(X_test, y_test)

print('test loss: ', loss)
print('test accuracy: ', accuracy)

结果还不错。。。

 9088/10000 [==========================>...] - ETA: 0s
 9536/10000 [===========================>..] - ETA: 0s
10000/10000 [==============================] - 1s 110us/step
test loss:  0.17950513743460178
test accuracy:  0.9495