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

win下的环境为keras2.16,运行lstm时报错

TypeError: softmax() got an unexpected keyword argument 'axis'
---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-7-ec539a8d38d4> in <module>()
      7 
      8 model.add(Dropout(0.2))#随机失活,避免过拟合,在训练的时候
----> 9 model.add(Dense(y.shape[1],activation="softmax"))#稠密层中定义激活函数,整体是一个softmax层,
     10 ####待查softmax层的原理,softmax的keras实现,
     11 #y.shape[1]是词典长度,activation是激活函数,Dense(y.shape[1],activation="softmax")是一个输出层,

 

解决方案:回退keras版本

keras 2.1.6的softmax没有axis这个参数了,于是,我回退了keras版本:

pip uninstall Keras

pip install keras==2.1

问题解决

 

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值