Как использовать часть данных? - PullRequest
0 голосов
/ 13 мая 2018

Я хочу использовать часть случайным образом из набора данных MNIST.Не могли бы вы мне помочь?Теперь выходная форма (т.е. Out) равна 60000, но я хочу получить около 2000:

import matplotlib.pyplot as plt
from keras.datasets import mnist

(x_train, y_train), (x_test, y_test) = mnist.load_data()
x_train = x_train.reshape(60000, 784) / 255
x_test = x_test.reshape(10000, 784) / 255

x_train.shape  # Out: (60000, 748)

1 Ответ

0 голосов
/ 13 мая 2018

Просто возьмите кусочек x_train:

new_x_train = x_train[:2000]

Если упорядочены данные в x_train (т. Е. Цифры класса 1, затем класса 2 и т. Д.), То вам следует сначала перемешатьданные и затем нарезать их на куски:

import numpy as np
indices = np.arange(x_train.shape[0])
np.random.shuffle(indices)
x_train = x_train[indices]

Подробнее о нарезке в нудистской документации .

...