Задать вопрос
@Andrei1penguin1

Почему Keras не принимает функцию потерь mse?

Доброго времени суток, есть задача детектирования объекта на изображении, соответственно, модель заканчивается слоем
Dense(4, activation="sigmoid")
Компилируется следующим образом:
model.compile(optimizer="rmsprop", loss="mse")
Но при попытке обучить её выскакивает следующая ошибка:
ValueError: A target array with shape (25, 200, 200, 3) was passed for an output of shape (None, 4) while using as loss `mean_squared_error`. This loss expects targets to have the same shape as the output.
Подскажите пожалуйста, как это можно исправить?
  • Вопрос задан
  • 54 просмотра
Подписаться 1 Простой 5 комментариев
Помогут разобраться в теме Все курсы
  • Нетология
    Fullstack-разработчик на Python + нейросети
    20 месяцев
    Далее
  • Skillfactory
    Профессия Python-разработчик
    9 месяцев
    Далее
  • SF Education
    Бэкенд-разработчик на Python
    3 месяца
    Далее
Пригласить эксперта
Ваш ответ на вопрос

Войдите, чтобы написать ответ

Похожие вопросы