关于TypeError: softmax() got an unexpected keyword argument 'axis'的解决方案

问题描述:在做Keras多层感知器识别手写体数字的时候,当导入

model.add(Dense(units=1000,input_dim=784,kernel_initializer='normal',activation='softmax'))

程序编译报错

Traceback (most recent call last):

  File "<ipython-input-26-4852e417c987>", line 1, in <module>
    runfile('C:/Users/lijunyi/Desktop/temp.py', wdir='C:/Users/lijunyi/Desktop')

  File "C:\Users\lijunyi\Anaconda3\envs\tensorflow\lib\site-packages\spyder\utils\site\sitecustomize.py", line 710, in runfile
    execfile(filename, namespace)

  File "C:\Users\lijunyi\Anaconda3\envs\tensorflow\lib\site-packages\spyder\utils\site\sitecustomize.py", line 101, in execfile
    exec(compile(f.read(), filename, 'exec'), namespace)

  File "C:/Users/lijunyi/Desktop/temp.py", line 65, in <module>
    model.add(Dense(units=10,kernel_initializer='normal',activation='softmax'))

  File "C:\Users\lijunyi\Anaconda3\envs\tensorflow\lib\site-packages\keras\engine\sequential.py", line 181, in add

  File "C:\Users\lijunyi\Anaconda3\envs\tensorflow\lib\site-packages\keras\engine\base_layer.py", line 457, in __call__

  File "C:\Users\lijunyi\Anaconda3\envs\tensorflow\lib\site-packages\keras\layers\core.py", line 883, in call

  File "C:\Users\lijunyi\Anaconda3\envs\tensorflow\lib\site-packages\keras\activations.py", line 31, in softmax
    def elu(x, alpha=1.0):

  File "C:\Users\lijunyi\Anaconda3\envs\tensorflow\lib\site-packages\keras\backend\tensorflow_backend.py", line 3231, in softmax
    x = _preprocess_conv2d_input(x, data_format)

TypeError: softmax() got an unexpected keyword argument 'axis'

解决办法

第一步:安装tensorflow1.5版本(也可以直接跳到第二步)

第二步:安装keras2.0.8版本

第三步:每次安装完以后不要直接运行程序,一定要重启核(如下图所示)

然后编译就不会报错了

 

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值