В Tensorflow при использовании предсказания, т. Е. «Print (предсказание [0] [0])», на какие значения [0] [0] действительно ссылаются?

Question

Просто быстро, я строю нейронную сеть для распознавания изображений, и я почти закончил. При тестировании прогноза моей модели на моих тестовых данных я выводю прогноз (очевидно), и в учебнике, которому я следую, используется выражение print(prediction[0][0]) (хотя разница в том, что в этом учебнике используется двоичный классификатор потерь).

Мне было просто интересно, что каждый [0] ссылается в конце оператора прогнозирования (который у меня есть как переменная, присвоенная model.predict(testX)), просто чтобы я мог легче поиграть со своими тестовыми данными и знать, что я ' Я выбираю. Я уверен, что это очень просто, но я только что сделал это. Любая помощь очень ценится!

Answer 1

Если вы хотите понять, что означает каждый [0], вы всегда можете просто напечатать их, то есть print(prediction[0]). Кроме того, вы можете использовать Jupyter Notebooks , чтобы возиться с этим материалом.

Но, чтобы ответить на ваш вопрос, model.predict возвращает list значений прогноза, а не одного прогноза, поэтому используется первый [0] (я предполагаю, что у вас был набор данных из один элемент), чтобы взять первое (и только в вашем случае) предсказание из списка. Затем второй [0] возвращает фактическое значение прогнозирования (единственное) из вашего списка значений прогнозирования