keras 一:mnist_mlp.py

mnist_mlp.py 在mnist数据集上训练一个多层感知机

'''Trains a simple deep NN on the MNIST dataset.

Gets to 98.40% test accuracy after 20 epochs
(there is *a lot* of margin for parameter tuning).
2 seconds per epoch on a K520 GPU.
'''

from __future__ import print_function

import keras
from keras.datasets import mnist
from keras.models import Sequential
#Dense 全连接层
#Dropout 对于神经网络按一定概率将其从网络中丢弃,解决过拟合问题
from keras.layers import Dense, Dropout
from keras.optimizers import RMSprop

#epochs 训练过程中数据被轮的次数,向前向后传播时数据所有批次的单次训练迭代
#one epoch = numbers of iterations = N =训练样本数量/batch_size
batch_size = 128
num_classes = 10
epochs = 20

# the data, shuffled and split between train and test sets
(x_train, y_train), (x_test, y_test) = mnist.load_data()

#输入数据维度(x_train.shape[0],x_train.shape[1]*x_train.shape[2])(图像数量,行×列)
x_train = x_train.reshape(60000, 784)
x_test = x_test.reshape
  • 1
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值