zoukankan      html  css  js  c++  java
  • 神经网络结构设计 不一定是第一层神经元是输入维度数

    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


  • 相关阅读:
    B树,B+树
    中断
    死锁
    无锁队列
    Cookie和Session
    分布式系统一致性
    c++ 标准库迭代器失效
    html5 app图片预加载
    html5 手机APP计算高度问题
    html5 750 REM JS换算方法
  • 原文地址:https://www.cnblogs.com/bonelee/p/9084741.html
Copyright © 2011-2022 走看看