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

出现这个问题，有几种解决办法，可以调低一下keras的版本，比如：

pip install keras==2.1

不过还有个更方便的方法，从错误可知softmax中不包含axis这个参数，那么把axis参数替换成dim就可以了。源代码是这样的：

def softmax(x, axis=-1):"""Softmax of a tensor.# Argumentsx: A tensor or variable.axis: The dimension softmax would be performed on.The default is -1 which indicates the last dimension.# ReturnsA tensor."""return tf.nn.softmax(x, axis=axis)

更改成这样：

def softmax(x, axis=-1):"""Softmax of a tensor.# Argumentsx: A tensor or variable.axis: The dimension softmax would be performed on.The default is -1 which indicates the last dimension.# ReturnsA tensor."""return tf.nn.softmax(x, dim=axis)

就是改了最后一行。

如果这篇文章帮助到了你，或者你有任何问题，欢迎扫码关注微信公众号：一刻AI 在后台留言即可，让我们一起学习一起进步！

转载于:https://www.cnblogs.com/lky-learning/p/10588965.html

TypeError: softmax() got an unexpected keyword argument 'axis'相关推荐

成功解决return tf.nn.softmax(x, axis=axis) TypeError: softmax() got an unexpected keyword argument 'axis
成功解决return tf.nn.softmax(x, axis=axis) TypeError: softmax() got an unexpected keyword argument 'axis ...
deep Learning ERROR(1):TypeError: softmax() got an unexpected keyword argument ‘axis‘
keras版本报错 [问题描述] [原因描述] [解决办法] 方法一:降低keras版本(之前用的2.2.4现在换为2.0.4) 方法二:修改问题中出现的描述(TypeError: softmax() ...
pycharm提示softmax() got an unexpected keyword argument 'axis'错误
今天帮师弟跑py程序,用的keras,我的环境是pycharm加anaconda(pycharm可以通过指定路径找到anaconda的配的环境,然后pycharm就不用在interpreter里面重新 ...
成功解决python\ops\seq2seq.py TypeError: ms_error() got an unexpected keyword argument 'labels'
成功解决python\ops\seq2seq.py TypeError: ms_error() got an unexpected keyword argument 'labels' 目录解决问题 ...
【Python异常】TypeError: __init__() got an unexpected keyword argument ‘size‘与gensim.models.word2vec 参数
文章目录一.遇到的异常: 1.TypeError: __init__() got an unexpected keyword argument 'size' 2.TypeError: __init_ ...
TypeError: __init__() got an unexpected keyword argument #34serialized_options #34
TypeError: __init__() got an unexpected keyword argument 'serialized_options' TypeError: __init__() ...
flask_sqlalchemy连接Mysql报TypeError: create_engine() got an unexpected keyword argument 'encoding'解决办法
最近在使用flask_sqlalchemy连接mysql,但是遇到一个很奇葩的问题,代码如下: #-*-coding:utf-8 -*-from flask_sqlalchemy import SQL ...
成功解决TypeError: read_excel() got an unexpected keyword argument ‘parse_cols or ‘sheetname‘
成功解决TypeError: read_excel() got an unexpected keyword argument 'parse_cols' TypeError: read_excel() ...
成功解决TypeError: take() got an unexpected keyword argument ‘fill_value‘
成功解决TypeError: take() got an unexpected keyword argument 'fill_value' 目录解决问题解决思路解决方法解决问题 Traceba ...

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

如果这篇文章帮助到了你，或者你有任何问题，欢迎扫码关注微信公众号：一刻AI 在后台留言即可，让我们一起学习一起进步！

TypeError: softmax() got an unexpected keyword argument 'axis'相关推荐

最新文章

热门文章