Как отсортировать набор данных в pytorch

Попробуйте ввести код.

Random rnd = new Random();
int month = rnd.Next(1, 13); // creates a number between 1 and 12
int dice = rnd.Next(1, 7); // creates a number between 1 and 6
int card = rnd.Next(52); // creates a number between 0 and 51
0
задан desertnaut 3 March 2019 в 00:36
поделиться

1 ответ

Не существует общего способа сделать это эффективно, так как класс набора данных реализует только методы __getitem__ и __len__ и не обязательно имеет какую-либо «хранимую» информацию о метках.

В случае класса набора данных MNIST , однако, вы можете отсортировать набор данных из списка меток.

Например, когда вы хотите перечислить индексы, которые имеют метку 5.

mnist = torchvision.datasets.mnist.MNIST("/")
labels = mnist.train_labels
fives = (labels == 5).nonzero()

0
ответ дан Chris Holland 3 March 2019 в 00:36
поделиться
Другие вопросы по тегам:

Похожие вопросы: