如何從 Keras 提供的 MNIST 數據集中只選擇特定數量的樣本？

Question

我目前正在使用 Keras 在 MNIST 數據集上訓練卷積神經網絡。 我正在使用格式加載數據集

(X_train, Y_train), (X_test, Y_test) = mnist.load_data()

但是為了減少對所有數據的迭代，我只想從每個 0-9 類中為X_train選擇前 10000 個樣本，類似地從Y_train 。 我怎樣才能做到這一點？

Answer 1

MNIST 數據集說它返回：

Return:

    2 tuples:
        X_train, X_test: uint8 array of grayscale image data with shape (nb_samples, 28, 28).
        y_train, y_test: uint8 array of digit labels (integers in range 0-9) with shape (nb_samples,).

所以你只需要切片你想保留的部分。 我相信 pandas/numpy 的語法類似於：

X_train = X_train[:10000,:,:]
X_test = X_test[:10000,:,:]
y_train = y_train[:10000]
y_test  = y_test[:10000]

Answer 2

from keras.datasets import mnist

(x_train, y_train), (x_test, y_test) = mnist.load_data()

x_train = x_train[:1000,:,:]
x_test = x_test[:500,:,:]
y_train = y_train[:1000]
y_test  = y_test[:500]


print(len(x_train))
print(len(y_train))
print(len(x_test))
print(len(y_test))

＃輸出

> 1000 
> 1000 
> 500 
> 500