Я пытаюсь создать мини-пакет из MNIST со всеми цифрами от 0 до 9. (10 элементов)
Я бы хотел не зацикливаться на всех элементах в моем редакторе меток, чтобы проверять числа по одному.
Какой самый простой подход для достижения этого?
Я подумал, что могу создать массив меток "all_digits" от 0 до 9, а затем сравнить его со своим списком mnist_labels "train_labels". (Одномерный массив - n элементов)
Я пытался получить матрицу проверки равенства всех пар. (п х 10)
Но я не мог напрямую использовать ==, и при этом не было транслируемой версии numpy.equal ().
У меня также не было четкого представления о том, что делать с матрицей впоследствии.
import numpy as np
train_labels = np.random.randint(0,10,100)
all_digits = np.arange(10)
# doing a difference for now
train_labels.reshape((-1,1)) - all_digits