Получение ошибки при создании наборов данных на основе класса набора данных CIFAR10

Question

Мне нужно создать 2 набора данных с одним набором данных, имеющим классы от 0 до 4, и другим, имеющим классы от 5 до 9 из CIFAR10 наборов данных, но я получаю эту ошибку: "boolean index did not match indexed array along dimension 1; dimension is 32 but corresponding boolean dimension is 1"

Вот чтоЯ пытался до сих пор

  import keras
  from keras.datasets import cifar10
  (x_train, y_train), (x_test, y_test) = cifar10.load_data()
  print('x_train shape:', x_train.shape)
  x_train shape: (50000, 32, 32, 3)
  Getting error at this point
  x_train = x_train[y_train < 5]

Answer 1

Распечатка y_train.shape дает (50000, 1). Чтобы правильно проиндексировать первое измерение x_train с помощью y_train, вам нужно убрать второе измерение.

x_train = x_train[y_train[:, 0] < 5]

запись [:, 0] означает, что все элементы по первому измерению возвращаются, а по второму - только первый элементразмерность.

x_train.shape теперь дает (25000, 32, 32, 3).