使用Keras减少尺寸

Question

我有一个MNIST手写数字的数据集，有257列。

1 : 256 - pixel values
257 - target

如何使用keras设计自动编码器以减少2维的输入。

我尝试了什么

encoding_dim = 32 
input_img = Input(shape=(256,))
encoded = Dense(encoding_dim, activation='relu')(input_img)
decoded = Dense(256, activation='sigmoid')(encoded)

# this model maps an input to its reconstruction
autoencoder = Model(input=input_img, output=decoded)
autoencoder.compile(optimizer='adadelta', loss='binary_crossentropy')
autoencoder.fit(X, X,
            nb_epoch=50,
            batch_size=256,
            shuffle=True)

错误

KeyError: '[318 327 ...] not in index

Answer 1

我可以通过使用numpy数组而不是pandas dataframe来解决这个问题。