csdn_black 2018-04-25 07:01 采纳率: 50%
浏览 9967
已结题

kears 运行一个最简单的demo报错

按照书上的代码如下:

 import numpy as np
import pandas as pd
from keras.utils import np_utils
np.random.seed(10)
from keras.models import Sequential
from keras.layers import Dense
from keras.datasets import mnist

#数据准备-------------------
(x_train_image, y_train_label), \
(x_test_image, y_test_label) = mnist.load_data()

x_Train =x_train_image.reshape(60000, 784).astype('float32')
x_Test = x_test_image.reshape(10000, 784).astype('float32')

#标准化
x_Train_normalize = x_Train/ 255
x_Test_normalize = x_Test/ 255

y_TrainOne_Hot = np_utils.to_categorical(y_train_label)
y_TestOne_Hot = np_utils.to_categorical(y_test_label)
print(x_Train_normalize)
print('sss')
print(y_TrainOne_Hot)

#建立模型-------------------

model = Sequential()
model.add(Dense(units=256,
                input_dim=784,
                kernel_initializer='normal',
                activation='relu'))

model.add(Dense(units=10,
                kernel_initializer='normal',
                activation='softmax'))

model.compile(loss='categorical_crossentropy',
              optimizer='adam', metrics=['accuracy'])

train_history =model.fit(x=x_Train_normalize,
                         y=y_TrainOne_Hot,validation_split=0.2,
                         epochs=10, batch_size=200,verbose=2)

报错:softmax() got an unexpected keyword argument 'axis'

详细的报错:
Traceback (most recent call last):
File "C:/Users/51530/PycharmProjects/ML/keras/K-MNIST/train.py", line 36, in
activation='softmax'))
File "E:\anaconda\lib\site-packages\keras\models.py", line 522, in add
output_tensor = layer(self.outputs[0])
File "E:\anaconda\lib\site-packages\keras\engine\topology.py", line 619, in call
output = self.call(inputs, **kwargs)
File "E:\anaconda\lib\site-packages\keras\layers\core.py", line 881, in call
output = self.activation(output)
File "E:\anaconda\lib\site-packages\keras\activations.py", line 29, in softmax
return K.softmax(x)
File "E:\anaconda\lib\site-packages\keras\backend\tensorflow_backend.py", line 2963, in softmax
return tf.nn.softmax(x, axis=axis)
TypeError: softmax() got an unexpected keyword argument 'axis'

有知道为什么会报错吗?先谢了

  • 写回答

4条回答 默认 最新

  • csdn_black 2018-04-25 07:44
    关注

    跪求大神帮助啊,感激不尽!!!!

    评论

报告相同问题?

悬赏问题

  • ¥15 有兄弟姐妹会用word插图功能制作类似citespace的图片吗?
  • ¥15 请教:如何用postman调用本地虚拟机区块链接上的合约?
  • ¥15 为什么使用javacv转封装rtsp为rtmp时出现如下问题:[h264 @ 000000004faf7500]no frame?
  • ¥15 乘性高斯噪声在深度学习网络中的应用
  • ¥15 运筹学排序问题中的在线排序
  • ¥15 关于docker部署flink集成hadoop的yarn,请教个问题 flink启动yarn-session.sh连不上hadoop,这个整了好几天一直不行,求帮忙看一下怎么解决
  • ¥15 深度学习根据CNN网络模型,搭建BP模型并训练MNIST数据集
  • ¥15 C++ 头文件/宏冲突问题解决
  • ¥15 用comsol模拟大气湍流通过底部加热(温度不同)的腔体
  • ¥50 安卓adb backup备份子用户应用数据失败