задача: to_categorical (керас) добавляет ось класса - PullRequest
0 голосов
/ 22 апреля 2020

У меня глубокий CNN и размер Xtrain (198,256,256,9), ie есть 198 тренировочных примеров, каждый с 9 каналами. У поезда есть (198,7). Мой прогноз - список чисел. Во-первых, возможно ли в этом случае использовать категориальную энтропию категории или более подходящую? Я получаю хорошие результаты с mse, и я пытался в течение ДНЕЙ заставить работать категорическую кроссентропию. Хорошо, вернемся к моим проблемам.

Когда я использую to_categorical (Ytrain, num_classes = 176), мой вывод теперь (198,7,176).

Мой последний слой в NN - Плотный (176). Так что ожидается двумерная цель. Однако, когда я горячо кодирую Ytrain, у меня остается трехмерная цель !!! Я знаю, что последнее измерение называется осью класса, но я не знаю, что делать дальше ...

отправить справку

...