как показать количество мнистов, когда вы оцениваете и знаете значение

Question

Я обучил модель тензорного потока, чем хочу попробовать, может ли он работать? Поэтому я выбираю число (a=train_data[0].reshape(1,28,28),b=test_data[1].reshape(1),model.evaluate(a,b)) и хочу узнать число числа (визуализация с помощью matplotlib).