神经网络结构设计 不一定是第一层神经元是输入维度数

def get_model(input_dim=33):
    # Build neural network   
    net = tflearn.input_data(shape=[None, input_dim])
    net = batch_normalization(net)
    #net = tflearn.fully_connected(net, input_dim) #去掉这层的话 精度95+,如果加上精度很难上95%
    net = tflearn.fully_connected(net, 128, activation='tanh')
    net = dropout(net, 0.5)
    net = tflearn.fully_connected(net, 2, activation='softmax')
    net = tflearn.regression(net, optimizer='sgd',
                     loss='categorical_crossentropy', name='target') #  optimizer='sgd'
    # Define model
    model = tflearn.DNN(net)
    #filename = 'CC_model_999.tflearn'
    #model.load(filename)
    #print filename + " loaded OK"
    return model


posted @   bonelee  阅读(985)  评论(0编辑  收藏  举报
编辑推荐:
· 记一次.NET内存居高不下排查解决与启示
· 探究高空视频全景AR技术的实现原理
· 理解Rust引用及其生命周期标识(上)
· 浏览器原生「磁吸」效果!Anchor Positioning 锚点定位神器解析
· 没有源码,如何修改代码逻辑?
阅读排行:
· 全程不用写代码,我用AI程序员写了一个飞机大战
· MongoDB 8.0这个新功能碉堡了,比商业数据库还牛
· 记一次.NET内存居高不下排查解决与启示
· 白话解读 Dapr 1.15:你的「微服务管家」又秀新绝活了
· DeepSeek 开源周回顾「GitHub 热点速览」
历史上的今天:
2017-05-24 LMDB中的mmap、Copy On Write、MVCC深入理解——讲得非常好,常来看看!
2017-05-24 golang LMDB入门例子——key range查询
2017-05-24 golang LMDB入门例子——尼玛,LMDB的文档真的是太少了
2017-05-24 golang import all 类似python import * 效果
2017-05-24 python lmdb demo 这接口和BDB一样恶心啊!
2017-05-24 ledisDB底层实现——本质上就是用leveldb这样的底层存储,和ssdb一样,meta里存的是hash、list等的元数据
2017-05-24 ssdb底层实现——ssdb底层是leveldb,leveldb根本上是skiplist(例如为存储多个list items,必然有多个item key,而非暴力string cat),用它来做redis的list和set等,势必在数据结构和算法层面上有诸多不适
点击右上角即可分享
微信分享提示