TypeError: softmax () получил неожиданный аргумент ключевого слова 'axis' - PullRequest
0 голосов
/ 09 июня 2018

Когда я использую это, оно не выдает никакой ошибки

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 Спасибо

Ответы [ 5 ]

0 голосов
/ 19 июня 2019

Причиной того, что эта ошибка является версией tenorflow и keras, является несоответствие.У меня есть эта проблема:

pip install tensorflow==1.5.0

Если вы не хотите понижать степень кера, tf 1.5.0 является первой версией, которая поддерживает softmax (axis = axis).

0 голосов
/ 12 июня 2019

Обновление tenorflow и keras до версии ниже решило мою проблему

pip install keras==2.1.6
pip install tensorflow==1.7.0
0 голосов
/ 15 июля 2018

Попробуйте:

import tensorflow as tf 

Затем добавьте слой softmax следующим образом:

model.add(Activation(tf.nn.softmax))
0 голосов
/ 04 апреля 2019

Вам необходимо установить TensorFlow.Вы можете сделать это с помощью одной из следующих команд:

pip install --upgrade tensorflow      # for Python 2.7
pip3 install --upgrade tensorflow     # for Python 3.n
0 голосов
/ 19 июня 2018

Обновите свои библиотеки tenSoflow и Keras до последних версий.Более низкие версии не поддерживают ось softmax.Убедитесь, что вы обновляете их в среде, в которой вы запускаете программу (очень важно).

...