ff_mnist.py 文件源码

python
阅读 26 收藏 0 点赞 0 评论 0

项目:deep_learning_ex 作者: zatonovo 项目源码 文件源码
def init_model():
    start_time = time.time()
    print 'Compiling Model ... '
    model = Sequential()
    model.add(Dense(500, input_dim=784))
    model.add(Activation('relu'))
    model.add(Dropout(0.4))
    model.add(Dense(300))
    model.add(Activation('relu'))
    model.add(Dropout(0.4))
    model.add(Dense(10))
    model.add(Activation('softmax'))

    rms = RMSprop()
    model.compile(loss='categorical_crossentropy', optimizer=rms,
      metrics=['accuracy'])
    print 'Model compiled in {0} seconds'.format(time.time() - start_time)
    return model
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号