Как в MNIST выбрать только определённую цифру (например, только нули)?

Question

[Александр Фёдоров]

Подскажите, пожалуйста, как (если вообще возможно) выбрать из базы MNIST для обучения только конкретную цифру?
Нужно выбрать только нули для обучения DCGAN.

Answer 1

[Данил]

from sklearn import datasets

# load only one class from 10 possible
mnist = datasets.load_digits(n_class=1)
X_train = mnist.images
y_train = mnist.target

Comments

[Александр Фёдоров]

Не знаете, есть ли аналогичный метод для keras?

[Данил]

 Похоже, что keras не предоставляет такой возможности. Можно написать свою обертку:

from keras.datasets import mnist
import numpy as np

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

def get_one_digit(digit):
  assert digit in range(10)
  X = []
  
  for label, value in zip(y_train, x_train):
    if label == digit:
      X.append(value)
    
  y = np.full(len(X), digit)
  
  return X, y

X_2, y_2 = get_one_digit(2)