如何編輯mnist數據集？

Question

我想在“mnist dataset”中有數字0到5。 如何在 python 上執行此操作？ 我嘗試用 numpy.delete 來解決這個問題，但沒有奏效。

Answer 1

假設您將圖像存儲在形狀(num_examples, num_pixels)的 numpy 數組中，並將標簽存儲在形狀(num_examples,)的數組中，您可以這樣做：

images = images[labels <= 5].copy()
labels = labels[labels <= 5].copy()

Answer 2

嘗試獲取與 range(6) 中的標簽關聯的圖像。 因此，您可以編寫一個類似於下面的 function：

images_5 = []
label_5 = []
for img, label in zip(images, labels):
    if label in range(6):
        images_5.append(img)
        labels_5.append(label)

不確定它是否會起作用，但我認為它應該！