1. ホーム
  2. python-2.7

[解決済み] TypeError: softmax() が予期しないキーワード引数 'axis' を受け取りました。

2022-01-26 03:24:39

質問

これを使ってもエラーは出ません。

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'))

というエラーが発生します。

TypeError: softmax() が予期しないキーワード引数 'axis' を受け取りました。

どうすればいいのでしょうか? python2を使っています。 ありがとうございます

どのように解決するのですか?

これを試してみてください。

import tensorflow as tf 

次に、このようにソフトマックスレイヤーを追加します。

model.add(Activation(tf.nn.softmax))