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

最新推荐文章于 2022-01-11 15:22:07 发布

旺仔的算法coding笔记

最新推荐文章于 2022-01-11 15:22:07 发布

阅读量735

点赞数

CC 4.0 BY-SA版权

分类专栏： python语法总结与bug积累

本文链接：https://blog.youkuaiyun.com/wangwangstone/article/details/86760411

python语法总结与bug积累专栏收录该内容

28 篇文章

订阅专栏

在使用Keras 2.1.6版本运行LSTM模型时遇到TypeError，错误信息显示softmax函数接收到未预期的参数'axis'。通过回退Keras版本至2.1解决了该问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

问题解决