按照书上的代码如下:
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'
有知道为什么会报错吗?先谢了