代码之家  ›  专栏  ›  技术社区  ›  Aakash aggarwal

类型错误:SoftMax()得到意外的关键字参数“axis”

  •  10
  • Aakash aggarwal  · 技术社区  · 6 年前

    当我使用这个时,它不会给出任何错误。

    out_layer = tf.add(tf.matmul(layer_4 , weights['out']) , biases['out'])
    out_layer = tf.nn.softmax(out_layer)
    

    但是当我用这个的时候

    model=Sequential()
    
    model.add(Dense(100, input_dim= n_dim, 
    activation='tanh',kernel_initializer='uniform'))
    keras.layers.core.Dropout(0.3, noise_shape=None, seed=None)
    
    model.add(Dense(50,input_dim=1000,activation='sigmoid'))
    keras.layers.core.Dropout(0.4, noise_shape=None, seed=None)
    
    model.add(Dense(15,input_dim=500,activation='sigmoid'))
    keras.layers.core.Dropout(0.2, noise_shape=None, seed=None)
    
    model.add(Dense(units=n_class))
    model.add(Activation('softmax'))
    

    我得到错误为

    类型错误:SoftMax()得到意外的关键字参数“axis”

    我该怎么办? 我在用蟒蛇2 谢谢

    5 回复  |  直到 5 年前
        1
  •  4
  •   Crisovet    6 年前

        2
  •  12
  •   petezurich rdelmar    6 年前

    import tensorflow as tf 
    

    model.add(Activation(tf.nn.softmax))
    
        3
  •  0
  •   Cody Gray    5 年前

    pip install --upgrade tensorflow      # for Python 2.7
    pip3 install --upgrade tensorflow     # for Python 3.n
    
        4
  •  0
  •   ybdesire    5 年前

    pip install keras==2.1.6
    pip install tensorflow==1.7.0
    
        5
  •  0
  •   Zzh.    5 年前

    pip install tensorflow==1.5.0