Как преобразовать массив numpy в 3 измерения для ввода в сверточный слой? - PullRequest
0 голосов
/ 16 февраля 2020

current situation of x_train array

У меня есть массив x_train, который состоит из 235 элементов, и каждый элемент представляет собой массив numpy строк 125 и 125 столбцов. Мне нужно, чтобы этот массив был в форме (235,125,125), чтобы соответствовать моей модели сверточной нейронной сети ((((аналогично случаю, когда нам нужно передать массив MNIST di git из (6000,28,28), где мы иметь 60000 сэмплов, каждый из которых массив 28x28)))

Может кто-нибудь сказать, как изменить форму массива x_train в (235,125,125)?

PS Я изначально преобразовал x_train из серии pandas в numpy массив ниже это снимок

1 Ответ

1 голос
/ 16 февраля 2020

Вам нужно использовать команду Numpy reshape:

x_train = x_train.reshape(x_train.shape[0], 125, 125, 1)

Взгляните на эту часть Учебник Keras 'MNIST CNN.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...