我正在尝试使用 Keras 实现本文中介绍的一个隐藏层模型。
这是我的代码:
from keras.models import Sequential
from keras.layers import Dense, Dropout, Flatten, Activation
from keras import optimizers
model = Sequential()
model.add(Dense(100, input_dim=9216))
model.add(Activation('relu'))
model.add(Dense(30))
model.compile(loss='mean_squared_error', optimizer=sgd, metrics=['accuracy'])
sgd = optimizers.SGD(lr=0.01, momentum=0.9, nesterov=True)
hist = model.fit(X_train, y_train, epochs=10, verbose=0, validation_split=0.2)
y_pred = model.predict_classes(X_valid)
X_train 形状为 (2140, 9216)
y_train 形状为 (2140, 30)
X_valid 形状为 (1783, 9216)
但y_valid形状是(1783,)。我试图理解为什么我没有得到 (1783, 30) 输出。我错过了什么吗?