python - 尝试运行自动编码器时出现 nolearn 错误

Question

我正在尝试使用以下命令运行一个简单的自动编码器nolearn：

import nolearn
from nolearn.dbn import DBN
from sklearn.cross_validation import train_test_split
data=np.load('doc_user_matrix.npy')
print (data.shape) #outputs: (10000,500)

(x_train, x_test, y_train, y_test)=train_test_split(data,data,test_size = 0.33)

hidden_layer=10

ae = DBN([x_train.shape[0], hidden_layer, x_train.shape[0]],
                learn_rates = 0.3,
                learn_rate_decays = 0.9,
                epochs = 10)

ae.fit(x_train, x_train)

出于某种原因，我遇到了这个错误：

ValueError：输入形状错误（10000、500）

谁能解释为什么会出现这个错误，以及如何解决它？

Answer 1

通常第二个维度对应于特征，而第一个维度对应于实例。尝试：

[x_train.shape[1], hidden_layer, x_train.shape[1]]