Как обозначить cnn в керасе и использовать метрики mse - PullRequest
0 голосов
/ 18 мая 2019

Я должен тренировать CNN, используя в качестве функции потерь mse.Я нормализовал метки между 0-1, но когда я тренируюсь, я получаю

ожидал, что плотность_2 будет иметь форму (1,), но получил массив с формой (7,)

Последний слой - x = Dense(1, activation='linear')(x)

, а когда я компилирую и тренируюсь:

model.compile(loss=keras.losses.mean_squared_error, optimizer=Adam(lr=0.001),metrics=['mse'])


model_details=model.fit(X_train, Y_train)

1 Ответ

0 голосов
/ 18 мая 2019

Я изменил размер матрицы Y_train до 1D вектора: каждый компонент вектора является меткой для соответствующего изображения.

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