Как изменить функцию потери тензорного потока с одного числа на массив - PullRequest
0 голосов
/ 23 октября 2018

Я работаю с модулем DNNRegressor компании Tensorflow для обучения 2-х скрытых многоуровневых нейронных сетей по 100 узлов каждый, чтобы отобразить входной слой, содержащий 9 параметров, и выходной слой, содержащий 3980 параметров.

Размер моегоУчебный набор составляет примерно 500 000 точек данных.До сих пор я достиг определенных успехов в обучении этой нейронной сети.Я работаю с функцией потерь по умолчанию, которая является среднеквадратичной ошибкой.Это одна переменная с плавающей запятой, и я хотел бы иметь функцию потерь, которая имеет тот же размер, что и выходной слой - 3980 плавающих значений (потому что выходной слой представляет собой спектр света).

Есть кто-нибудьрассматривал / имел дело с такой ситуацией?

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