Как я могу использовать поиск по сетке для настройки гиперпараметра, который является частью потери? - PullRequest
0 голосов
/ 27 апреля 2020

У меня есть следующая проблема: Вот как я компилирую модель.

model.compile(loss=custom_loss_wrapper(model.input, std, mean),
              optimizer=optimizer,
              metrics=['mean_absolute_error', 'mean_squared_error'],
              experimental_run_tf_function=False)

Я использую пользовательскую функцию потерь, которая состоит из обычного MSE между y_true и y_pred и штрафной части, суммируемой на MSE и масштабируется с гиперпараметром альфа.

Чтобы быть более точным, у меня есть потери, которые: L() = MSE() + (1/alpha) * || something ||

Я хотел бы использовать поиск по сетке для настройки этого параметра альфа, но я не уверен, как мне это сделать, поскольку это часть пользовательской регуляризации, а не обычной.

Любое предложение?

Заранее спасибо !!

...